update

intel · Jun 21, 2024 · ee262e6 · ee262e6
1 parent d2b310a
commit ee262e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_on_ray/finetune/finetune.py b/llm_on_ray/finetune/finetune.py
@@ -374,7 +374,7 @@ def train_func(config: Dict[str, Any]):
         result = trainer.train(resume_from_checkpoint=training_args.resume_from_checkpoint)
         trainer.save_model()
         metrics = result.metrics
-        metrics["throughput"] = len(tokenized_dataset['train']) / metrics["train_runtime"]
+        metrics["throughput"] = len(tokenized_dataset["train"]) / metrics["train_runtime"]
         trainer.log_metrics("train", metrics)
         common.logger.info("train finish")