From e410d2e1c902c747200e0231377c7daed54a8db7 Mon Sep 17 00:00:00 2001 From: dorren Date: Mon, 17 Oct 2022 19:16:02 +0800 Subject: [PATCH] update tipc config --- test_tipc/configs/rec_d28_can/rec_d28_can.yml | 8 ++++---- test_tipc/configs/rec_d28_can/train_infer_python.txt | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) 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 2149100d..92917c20 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 1794e78c..731d327c 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 -- GitLab