Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
harborn committed Dec 27, 2023
1 parent 5e3cc83 commit ac1c448
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions finetune/finetune.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,6 +108,7 @@ def train_func(config: Dict[str, Any]):
trainer = common.trainer.Trainer.registory.get("DefaultTrainer")(config = {
"num_train_epochs": config["Training"]["epochs"],
"max_train_step": config["Training"].get("max_train_steps", None),
"log_step": 1,
"output": config["General"]["output_dir"],
"dataprocesser": {
"type": "GeneralProcesser",
Expand Down

0 comments on commit ac1c448

Please sign in to comment.