From 9b8c0ef39715eeca680db41726fbac566a616e50 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Sun, 19 Dec 2021 07:46:56 +0000 Subject: [PATCH] fix yml --- configs/kie/kie_unet_sdmgr.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/configs/kie/kie_unet_sdmgr.yml b/configs/kie/kie_unet_sdmgr.yml index 55d07be2..a6968aaa 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 -- GitLab