From 12a7c28573ebd871d3f86ac33eca9e803eb3363e Mon Sep 17 00:00:00 2001 From: Xiaohang Zhan Date: Sun, 19 Jul 2020 18:08:00 +0800 Subject: [PATCH] Update dist_train_linear.sh --- benchmarks/dist_train_linear.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/dist_train_linear.sh b/benchmarks/dist_train_linear.sh index 65821ae2..ce1c4224 100755 --- 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