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 c2295af96629d642ffd8f73dbff500680af60b90..13eda8481cad8ca308cd0629214b52146c3ebf13 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