yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-12 16:39:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Typo

Browse Source
This commit is contained in:
Jake Poznanski 2024-09-26 19:57:53 +00:00
parent cf1aa0176e
commit 49efa5cb40
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pdelfin/train/train.py
View File

@ -198,8 +198,8 @@ def run_train(config: TrainConfig):
trainer = Trainer( trainer = Trainer(
model=model, model=model,
args=training_args, args=training_args,
train_dataset=formatted_dataset["train"], train_dataset=train_ds,
eval_dataset=formatted_dataset["validation"], # pyright: ignore eval_dataset=validation_ds,
tokenizer=processor.tokenizer, tokenizer=processor.tokenizer,
#Collator is not needed as we are doing batch size 1 for now... #Collator is not needed as we are doing batch size 1 for now...
#data_collator=collator, #data_collator=collator,