diff --git a/cv/classification/resnet50/pytorch/README.md b/cv/classification/resnet50/pytorch/README.md index 6e4e5da3810c18cf953c5271dee03cff3877443e..5a4f9318ca88b8ad6fb6135e5a595322c691cc2d 100644 --- a/cv/classification/resnet50/pytorch/README.md +++ b/cv/classification/resnet50/pytorch/README.md @@ -4,7 +4,7 @@ Residual Networks, or ResNets, learn residual functions with reference to the la ## Step 1: Installing ```bash -pip3 install -r requirements.txt +pip3 install torch torchvision ``` :beers: Done! diff --git a/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh b/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh index 68e3ff1871ce5cf1c88eb552cfa1a5ad8b421044..4fa6953dae786cd10ae4ef1594593b04b8f61b49 100755 --- a/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh +++ b/cv/classification/vgg/pytorch/train_vgg16_amp_dist.sh @@ -26,5 +26,5 @@ fi cd ${ROOT_DIR} python3 $PYTHONARG ${ROOT_DIR}/run_train.py \ --model vgg16 --dali --dali-cpu --data-path $DATA_PATH \ - --opt fused_sgd --batch-size 300 --lr 1e-2 \ + --opt fused_sgd --batch-size 200 --lr 1e-2 \ --amp --nhwc "$@"