weight_decay fix
This commit is contained in:
parent
12b169158f
commit
fb3bdb9372
|
@ -2,7 +2,7 @@
|
||||||
FROM nvcr.io/nvidia/pytorch:19.07-py3
|
FROM nvcr.io/nvidia/pytorch:19.07-py3
|
||||||
|
|
||||||
# Install dependencies
|
# Install dependencies
|
||||||
RUN pip3 install -U -r requirements.txt
|
RUN pip install -U -r requirements.txt
|
||||||
|
|
||||||
# Move file into container
|
# Move file into container
|
||||||
# RUN mv 1047.tif ./1047.tif
|
# RUN mv 1047.tif ./1047.tif
|
||||||
|
@ -14,7 +14,7 @@ RUN pip3 install -U -r requirements.txt
|
||||||
# --------------------------------------------------- Extras Below ---------------------------------------------------
|
# --------------------------------------------------- Extras Below ---------------------------------------------------
|
||||||
|
|
||||||
# Build container
|
# Build container
|
||||||
# sudo docker image prune -a && sudo docker build -t friendlyhello . && sudo docker tag friendlyhello ultralytics/yolov3:v0
|
# sudo docker image prune -af -y && sudo docker build -t friendlyhello . && sudo docker tag friendlyhello ultralytics/yolov3:v0
|
||||||
|
|
||||||
# Run container
|
# Run container
|
||||||
# time sudo docker run -it --memory=8g --cpus=4 ultralytics/yolov3:v0 bash -c './run.sh /1047.tif /tmp && cat /tmp/1047.tif.txt'
|
# time sudo docker run -it --memory=8g --cpus=4 ultralytics/yolov3:v0 bash -c './run.sh /1047.tif /tmp && cat /tmp/1047.tif.txt'
|
||||||
|
|
Loading…
Reference in New Issue