From a4d9da9aeaffa4ea1610c0d554997c924cc48038 Mon Sep 17 00:00:00 2001 From: Meeeeee6623 Date: Tue, 31 Aug 2021 20:53:08 -0400 Subject: [PATCH] Add print to script --- TensorFlow/Classification/ConvNets/tf_by_epoch.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/TensorFlow/Classification/ConvNets/tf_by_epoch.sh b/TensorFlow/Classification/ConvNets/tf_by_epoch.sh index 905a8aa5..8c82eaec 100644 --- a/TensorFlow/Classification/ConvNets/tf_by_epoch.sh +++ b/TensorFlow/Classification/ConvNets/tf_by_epoch.sh @@ -1,5 +1,7 @@ for (( i = 0; i < 91; i++ )); do mkdir /data/tfrecords/epoch_"${i}"/ + echo "Created Folder epoch_${i}" mpiexec --allow-run-as-root --bind-to socket -np 2 python3 main.py --arch=resnet50 --mode=train_and_evaluate --num_iter=i --batch_size=192 --warmup_steps=100 --cosine_lr --label_smoothing 0.1 --lr_init=0.256 --lr_warmup_epochs=8 --momentum=0.875 --weight_decay=3.0517578125e-05 --model_dir=/data/tfrecords/amp_model --data_dir=/data/tfrecords/tfrecords --data_idx_dir=/data/tfrecords/dali_idx --results_dir=/data/tfrecords/epoch_"${i}"/results --export_dir=/data/tfrecords/best_model --weight_init=fan_in --amp --log_filename epoch_"${i}".json + echo "Epoch ${i} done" done