maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

add tensorflow script for cifar100

Browse source
This commit is contained in:
Meeeeee6623 2021-09-17 15:38:14 -04:00
parent a0eb2c15c8
commit 4cc3db0d8b
2 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
TensorFlow/Classification/ConvNets/tf_3_runs.sh
View file

@ -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

14
TensorFlow/Classification/ConvNets/tf_cifar100_by_epoch.sh Normal file
View file

@ -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