Initial commit

2024-02-11 10:44:42 +00:00
parent f057490bdb
commit b4962bfa98
1 changed files with 1 additions and 1 deletions
--- 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: