YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
三洋三洋
2024-02-11 10:44:42 +00:00
parent f057490bdb
commit b4962bfa98
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/ocr_model/train/train.py
View File

@@ -96,7 +96,7 @@ if __name__ == '__main__':
#+) #+)
enable_train = True enable_train = True
enable_evaluate = True enable_evaluate = False
if enable_train: if enable_train:
train(model, tokenizer, train_dataset, eval_dataset, collate_fn_with_tokenizer) train(model, tokenizer, train_dataset, eval_dataset, collate_fn_with_tokenizer)
if enable_evaluate and len(eval_dataset) > 0: if enable_evaluate and len(eval_dataset) > 0: