add by epoch script to show tensorflow accuracy

This commit is contained in:
Meeeeee6623 2021-08-31 20:50:59 -04:00
parent 01bbec9c4e
commit 575400b261

View file

@ -0,0 +1,5 @@
for (( i = 0; i < 91; i++ )); do
mkdir /data/tfrecords/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
done