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: