Commit 820546d6 authored by Kapsner's avatar Kapsner
Browse files

deps: torchmetrics from df to req.txt

parent 0e24b31f
......@@ -47,6 +47,3 @@ RUN mkdir ${det_data} \
WORKDIR /opt/code/nndet
COPY . .
RUN FORCE_CUDA=1 pip install -v -e .
# torchmetrics=='0.8.1' fails, so downgrade here
RUN yes | pip install torchmetrics=='0.7.3'
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment