From 06a434bcf916407a575cfcae4288dfb3ba7c408f Mon Sep 17 00:00:00 2001 From: tink2123 Date: Tue, 26 Jan 2021 15:57:37 +0800 Subject: [PATCH] fix yml --- .../multi_language/rec_en_number_lite_train.yml | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/configs/rec/multi_language/rec_en_number_lite_train.yml b/configs/rec/multi_language/rec_en_number_lite_train.yml index c2295af9..13eda848 100644 --- a/configs/rec/multi_language/rec_en_number_lite_train.yml +++ b/configs/rec/multi_language/rec_en_number_lite_train.yml @@ -1,5 +1,5 @@ Global: - use_gpu: False + use_gpu: True epoch_num: 500 log_smooth_window: 20 print_batch_step: 10 @@ -63,8 +63,8 @@ Metric: Train: dataset: name: SimpleDataSet - data_dir: ./train_data/ic15_data/ - label_file_list: ["./train_data/ic15_data/rec_gt_test.txt"] + data_dir: ./train_data/ + label_file_list: ["./train_data/train_list.txt"] transforms: - DecodeImage: # load image img_mode: BGR @@ -77,15 +77,15 @@ Train: keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order loader: shuffle: True - batch_size_per_card: 1 + batch_size_per_card: 256 drop_last: True - num_workers: 1 + num_workers: 8 Eval: dataset: name: SimpleDataSet - data_dir: ./train_data/ic15_data/ - label_file_list: ["./train_data/ic15_data/rec_gt_test.txt"] + data_dir: ./train_data/ + label_file_list: ["./train_data/eval_list.txt"] transforms: - DecodeImage: # load image img_mode: BGR -- GitLab