diff --git a/run.sh b/run.sh index aedc4ed..9f6265b 100644 --- a/run.sh +++ b/run.sh @@ -4,4 +4,4 @@ python classification_vgg.py python create_DDT_dataset.py python DDT/generate_box_imagenet_crop.py python tools/train.py -c svtr_large_train_stn.yml # Distributed Training: python -m paddle.distributed.launch --gpus '0,1,2,3' -c svtr_large_train_stn.yml -python tools/infer_rec.py -c svtr_large_train_stn.yml -o Global.pretrained_model=svtr_large_stn/best_accuracy \ No newline at end of file +python tools/infer_rec.py -c svtr_large_train_stn.yml -o Global.pretrained_model=svtr_large_stn/latest