diff --git a/finetune/finetune.py b/finetune/finetune.py index c2c8774ec..825c91d72 100644 --- a/finetune/finetune.py +++ b/finetune/finetune.py @@ -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",