Update dist_train_linear.sh

2020-07-19 18:08:00 +08:00 · 2020-07-19 18:08:00 +08:00 · 12a7c28573
parent e928ec979b
commit 12a7c28573
1 changed files with 1 additions and 1 deletions
--- a/benchmarks/dist_train_linear.sh
+++ b/benchmarks/dist_train_linear.sh
@ -6,7 +6,7 @@ set -x
 CFG=$1 # use cfgs under "configs/benchmarks/linear_classification/"
 PRETRAIN=$2
 PY_ARGS=${@:3} # --resume_from --deterministic
-GPUS=8 # When changing GPUS, please also change imgs_per_batch in the config file accordingly to ensure the total batch size is 256.
+GPUS=8 # When changing GPUS, please also change imgs_per_gpu in the config file accordingly to ensure the total batch size is 256.
 PORT=${PORT:-29500}

 if [ "$CFG" == "" ] || [ "$PRETRAIN" == "" ]; then