Skip to content

Commit

Permalink
fix: trainer argument
Browse files Browse the repository at this point in the history
  • Loading branch information
zzangsu committed Jun 8, 2024
1 parent 2727b66 commit 2bbf1da
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion autosink_model_training/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 콜백 추가
)

Expand Down

0 comments on commit 2bbf1da

Please sign in to comment.