revert-1929-fix_typo
tink2123 4 years ago
parent edeb12b1e0
commit 06a434bcf9

@ -1,5 +1,5 @@
Global: Global:
use_gpu: False use_gpu: True
epoch_num: 500 epoch_num: 500
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
@ -63,8 +63,8 @@ Metric:
Train: Train:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./train_data/ic15_data/ data_dir: ./train_data/
label_file_list: ["./train_data/ic15_data/rec_gt_test.txt"] label_file_list: ["./train_data/train_list.txt"]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR
@ -77,15 +77,15 @@ Train:
keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
loader: loader:
shuffle: True shuffle: True
batch_size_per_card: 1 batch_size_per_card: 256
drop_last: True drop_last: True
num_workers: 1 num_workers: 8
Eval: Eval:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./train_data/ic15_data/ data_dir: ./train_data/
label_file_list: ["./train_data/ic15_data/rec_gt_test.txt"] label_file_list: ["./train_data/eval_list.txt"]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR

Loading…
Cancel
Save