s444415/donut
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fixes, missing args

Browse Source
This commit is contained in:
s444415 2023-01-04 11:33:50 +01:00
parent c263335830
commit fe0b8c0397
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
donut-train.py
View File

@ -64,7 +64,7 @@ def main(config, hug_token):
login(hug_token, True)
model_module = DonutModelPLModule(config.train_config.toDict(), processor, model)
model_module = DonutModelPLModule(config.train_config.toDict(), processor, model, max_length=config.max_length, train_dataloader=train_dataloader, val_dataloader=val_dataloader)
wandb_logger = WandbLogger(project="Donut", name=config.wandb_test_name)
@ -90,7 +90,7 @@ def main(config, hug_token):
plugins=custom_ckpt,
num_sanity_val_steps=0,
logger=wandb_logger,
callbacks=[PushToHubCallback(), checkpoint_callback],
callbacks=[PushToHubCallback(output_model_path=config.output_model_path), checkpoint_callback],
)
trainer.fit(model_module)