From 2bbf1daf36a6f63c5f7359873718984ed114c537 Mon Sep 17 00:00:00 2001 From: subin Date: Sat, 8 Jun 2024 15:01:10 +0900 Subject: [PATCH] fix: trainer argument --- autosink_model_training/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autosink_model_training/train.py b/autosink_model_training/train.py index 1fa939f..d82f3d1 100644 --- a/autosink_model_training/train.py +++ b/autosink_model_training/train.py @@ -54,7 +54,7 @@ def get_trainer(self): args=self.training_args, train_dataset=self.train_dataset, eval_dataset=self.eval_dataset, - compute_metrics=self.get_compute_metrics_fn('accuracy'), + compute_metrics=get_compute_metrics_fn('accuracy'), # callbacks=[MyWandbCallback()] # W&B 콜백 추가 )