weight_decay fix

This commit is contained in:
Glenn Jocher 2019-08-29 20:13:29 +02:00
parent dc9095ac5b
commit 6c4a855d98
1 changed files with 8 additions and 4 deletions

View File

@ -8,11 +8,11 @@ WORKDIR /usr/src/app
# Copy contents # Copy contents
COPY . /usr/src/app COPY . /usr/src/app
# Install dependencies # Install dependencies (pip or conda)
# RUN pip install -U -r requirements.txt # RUN pip install -U -r requirements.txt
RUN conda update -n base -c defaults conda # RUN conda update -n base -c defaults conda
RUN conda install -y -c anaconda future numpy opencv matplotlib tqdm pillow # RUN conda install -y -c anaconda future numpy opencv matplotlib tqdm pillow
RUN conda install -y -c conda-forge scikit-image tensorboard pycocotools # RUN conda install -y -c conda-forge scikit-image tensorboard pycocotools
# conda install pytorch torchvision -c pytorch # conda install pytorch torchvision -c pytorch
# Move model into container # Move model into container
@ -22,10 +22,14 @@ RUN conda install -y -c conda-forge scikit-image tensorboard pycocotools
# --------------------------------------------------- Extras Below --------------------------------------------------- # --------------------------------------------------- Extras Below ---------------------------------------------------
# Build container # Build container
# rm -rf yolov3 # Warning: remove existing
# git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 detect.py
# sudo docker image prune -af -y && 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'
# time sudo docker run -it --memory=8g --cpus=4 ultralytics/yolov3:v0 bash -c 'python3 detect.py'
# Push container to https://hub.docker.com/u/ultralytics # Push container to https://hub.docker.com/u/ultralytics
# sudo docker push ultralytics/xview:v30 # sudo docker push ultralytics/xview:v30