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

Add print to script

Browse source
This commit is contained in:
Meeeeee6623 2021-08-31 20:53:08 -04:00
parent 575400b261
commit a4d9da9aea
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

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