提交 06a434bc 编写于 作者: T tink2123

fix yml

上级 edeb12b1
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: ...@@ -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: ...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册