diff --git a/configs/kie/kie_unet_sdmgr.yml b/configs/kie/kie_unet_sdmgr.yml index 55d07be26c11b1542c78a71bd9e79d4509637652..a6968aaa3aa7a717a848416efc5ccc567f774b4d 100644 --- a/configs/kie/kie_unet_sdmgr.yml +++ b/configs/kie/kie_unet_sdmgr.yml @@ -1,6 +1,6 @@ Global: use_gpu: True - epoch_num: 300 + epoch_num: 60 log_smooth_window: 20 print_batch_step: 50 save_model_dir: ./output/kie_5/ @@ -13,7 +13,7 @@ Global: # you should set load_static_weights as False. load_static_weights: False cal_metric_during_train: False - pretrained_model: ./output/kie_4/best_accuracy + pretrained_model: checkpoints: save_inference_dir: use_visualdl: False @@ -108,4 +108,4 @@ Eval: shuffle: False drop_last: False batch_size_per_card: 1 # must be 1 - num_workers: 4 \ No newline at end of file + num_workers: 4