"src/git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "63eda0fe4252d66b046de833ef37862a2a898bf2"
Commit 06a434bc authored by tink2123's avatar tink2123
Browse files

fix yml

parent edeb12b1
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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment