Skip to content

Commit 00bde50

Browse files
peri044peri044
committed
Remove folder for QAT
1 parent ed7831d commit 00bde50

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

‎TensorFlow/Classification/ConvNets/resnet50v1.5/training/GPU1_RN50_QAT.sh‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
# This script does Quantization aware training of Resnet-50 by finetuning on the pre-trained model using 1 GPU and a batch size of 32.
2+
# Usage ./GPU1_RN50_QAT.sh <path to the pre-trained model> <path to dataset> <path to results directory>
3+
4+
python main.py --mode=train_and_evaluate --batch_size=32 --lr_warmup_epochs=1 --quantize --symmetric --use_qdq --label_smoothing 0.1 --lr_init=0.00005 --momentum=0.875 --weight_decay=3.0517578125e-05 --finetune_checkpoint=$1 --data_dir=$2 --results_dir=$3 --num_iter 10 --data_format NHWC

0 commit comments

Comments
 (0)