weight_decay fix
This commit is contained in:
parent
3c56c07f1e
commit
fffb7fb992
|
@ -1,5 +1,5 @@
|
|||
# Start from Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
||||
FROM nvcr.io/nvidia/pytorch:19.07-py3
|
||||
FROM nvcr.io/nvidia/pytorch:19.08-py3
|
||||
|
||||
# Create working directory
|
||||
RUN mkdir -p /usr/src/app
|
||||
|
|
Loading…
Reference in New Issue