diff --git a/test_tipc/configs/rec_d28_can/rec_d28_can.yml b/test_tipc/configs/rec_d28_can/rec_d28_can.yml index 2149100da556057e2dda967109d6e59d44dc1c42..92917c20d6c8186204b3b142dbbaeb3aafbb5edb 100644 --- a/test_tipc/configs/rec_d28_can/rec_d28_can.yml +++ b/test_tipc/configs/rec_d28_can/rec_d28_can.yml @@ -75,7 +75,7 @@ Metric: Train: dataset: name: SimpleDataSet - data_dir: ./train_data/CROHME/training/images/ + data_dir: ./train_data/CROHME_lite/training/images/ transforms: - DecodeImage: channel_first: False @@ -90,7 +90,7 @@ Train: lower: False - KeepKeys: keep_keys: ['image', 'label'] - label_file_list: ["./train_data/CROHME/training/labels.txt"] + label_file_list: ["./train_data/CROHME_lite/training/labels.txt"] loader: shuffle: True batch_size_per_card: 8 @@ -101,7 +101,7 @@ Train: Eval: dataset: name: SimpleDataSet - data_dir: ./train_data/CROHME/evaluation/images/ + data_dir: ./train_data/CROHME_lite/evaluation/images/ transforms: - DecodeImage: channel_first: False @@ -116,7 +116,7 @@ Eval: lower: False - KeepKeys: keep_keys: ['image', 'label'] - label_file_list: ["./train_data/CROHME/evaluation/labels.txt"] + label_file_list: ["./train_data/CROHME_lite/evaluation/labels.txt"] loader: shuffle: False drop_last: False diff --git a/test_tipc/configs/rec_d28_can/train_infer_python.txt b/test_tipc/configs/rec_d28_can/train_infer_python.txt index 1794e78cfa82a27dac058c50876f774d71b728ea..731d327cd085b41a6bade9b7092dda7b2de9d9f9 100644 --- a/test_tipc/configs/rec_d28_can/train_infer_python.txt +++ b/test_tipc/configs/rec_d28_can/train_infer_python.txt @@ -1,7 +1,7 @@ ===========================train_params=========================== model_name:rec_d28_can -python:python -gpu_list:0|0 +python:python3.7 +gpu_list:0|0,1 Global.use_gpu:True|True Global.auto_cast:null Global.epoch_num:lite_train_lite_infer=2|whole_train_whole_infer=240