From ac1c448bf96ab833bc151c4fd5e3c976dcc99712 Mon Sep 17 00:00:00 2001 From: "Wu, Gangsheng" Date: Wed, 27 Dec 2023 11:45:52 +0000 Subject: [PATCH] update --- finetune/finetune.py | 1 + 1 file changed, 1 insertion(+) 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",