weight_decay fix

This commit is contained in:
Glenn Jocher 2019-08-31 13:39:14 +02:00
parent 3c56c07f1e
commit fffb7fb992
1 changed files with 1 additions and 1 deletions

View File

@ -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