From 4cc3db0d8b8eafd5ba48f227bf57a611a754415f Mon Sep 17 00:00:00 2001 From: Meeeeee6623 Date: Fri, 17 Sep 2021 15:38:14 -0400 Subject: [PATCH] add tensorflow script for cifar100 --- TensorFlow/Classification/ConvNets/tf_3_runs.sh | 2 +- .../ConvNets/tf_cifar100_by_epoch.sh | 14 ++++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) create mode 100644 TensorFlow/Classification/ConvNets/tf_cifar100_by_epoch.sh diff --git a/TensorFlow/Classification/ConvNets/tf_3_runs.sh b/TensorFlow/Classification/ConvNets/tf_3_runs.sh index 44239452..6df2621c 100644 --- a/TensorFlow/Classification/ConvNets/tf_3_runs.sh +++ b/TensorFlow/Classification/ConvNets/tf_3_runs.sh @@ -1,4 +1,4 @@ -# nvidia-docker run --rm -it --ipc=host -v /work/chauhans/cifar100:/cifar100 rn50_tf1 +# nvidia-docker run --rm -it --ipc=host -v /work/chauhans/imagenet:/imagenet rn50_tf1 for (( n = 0; i < 3; n++ )); do for (( i = 0; i < 91; i++ )); do diff --git a/TensorFlow/Classification/ConvNets/tf_cifar100_by_epoch.sh b/TensorFlow/Classification/ConvNets/tf_cifar100_by_epoch.sh new file mode 100644 index 00000000..be82904a --- /dev/null +++ b/TensorFlow/Classification/ConvNets/tf_cifar100_by_epoch.sh @@ -0,0 +1,14 @@ +# nvidia-docker run --rm -it --ipc=host -v /work/chauhans/cifar100:/cifar100 rn50_tf1 + +for (( n = 0; i < 1; n++ )); do + for (( i = 0; i < 91; i++ )); do + mkdir /imagenet/TensorFlow/run_"${n}"/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=0 --lr_warmup_epochs=0 --model_dir=/imagenet/TensorFlow/model \ + --data_dir=/imagenet/tfrecords --results_dir=/imagenet/TensorFlow/run_"${n}"/epoch_"${i}"/ \ + --export_dir=/imagenet/TensorFlow/model --weight_init=fan_in --amp --log_filename run_"${n}"epoch_"${i}".json + echo "Epoch ${i} done" + done + echo "Run ${n} done" +done