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
...@@ -16,4 +16,5 @@ hydra-core>=1.1.0 ...@@ -16,4 +16,5 @@ hydra-core>=1.1.0
mlflow mlflow
GitPython GitPython
matplotlib matplotlib
seaborn seaborn
\ No newline at end of file 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