This commit is contained in:
tink2123 2021-01-26 15:57:37 +08:00
parent edeb12b1e0
commit 06a434bcf9
1 changed files with 7 additions and 7 deletions

View File

@ -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