Skip to content

GitLab

Commit 87bbe588 authored by Kapsner's avatar Kapsner
Browse files

fix: downgrading torchmetrics due to import error

parent cde84743
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
Dockerfile
View file @ 87bbe588
......@@ -47,3 +47,6 @@ 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