From b4962bfa9873496b6a7582204a26339ac044c161 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=89=E6=B4=8B=E4=B8=89=E6=B4=8B?= <1258009915@qq.com> Date: Sun, 11 Feb 2024 10:44:42 +0000 Subject: [PATCH] Initial commit --- src/models/ocr_model/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/models/ocr_model/train/train.py b/src/models/ocr_model/train/train.py index af0c7a1..0e2979d 100644 --- a/src/models/ocr_model/train/train.py +++ b/src/models/ocr_model/train/train.py @@ -96,7 +96,7 @@ if __name__ == '__main__': #+) enable_train = True - enable_evaluate = True + enable_evaluate = False if enable_train: train(model, tokenizer, train_dataset, eval_dataset, collate_fn_with_tokenizer) if enable_evaluate and len(eval_dataset) > 0: