diff --git a/config.yaml b/config.yaml index 56de924..22c8edd 100644 --- a/config.yaml +++ b/config.yaml @@ -30,13 +30,13 @@ gradient_clip: True use_label_mask: False # 训练数据 -train_image_path: 'dataset/CROHME/train_images.pkl' -train_label_path: 'dataset/CROHME/train_labels.txt' +train_image_path: 'datasets/CROHME/train_images.pkl' +train_label_path: 'datasets/CROHME/train_labels.txt' -eval_image_path1: 'dataset/CROHME/14_test_images.pkl' -eval_label_path1: 'dataset/CROHME/14_test_labels.txt' +eval_image_path1: 'datasets/CROHME/14_test_images.pkl' +eval_label_path1: 'datasets/CROHME/14_test_labels.txt' -word_path: 'dataset/CROHME/words_dict.txt' +word_path: 'datasets/CROHME/words_dict.txt' # collate_fn collate_fn: collate_fn