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} \ ...@@ -47,6 +47,3 @@ RUN mkdir ${det_data} \
WORKDIR /opt/code/nndet WORKDIR /opt/code/nndet
COPY . . COPY . .
RUN FORCE_CUDA=1 pip install -v -e . 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
...@@ -17,3 +17,4 @@ mlflow ...@@ -17,3 +17,4 @@ mlflow
GitPython GitPython
matplotlib matplotlib
seaborn seaborn
torchmetrics>=0.7.0,<=0.7.3
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