add by epoch script to show tensorflow accuracy
This commit is contained in:
parent
01bbec9c4e
commit
575400b261
5
TensorFlow/Classification/ConvNets/tf_by_epoch.sh
Normal file
5
TensorFlow/Classification/ConvNets/tf_by_epoch.sh
Normal 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
|
||||
|
Loading…
Reference in a new issue