ctheodoris hchen725 commited on
Commit
dc23c5f
1 Parent(s): 63275a8

Update trainer output dir (#427)

Browse files

- Update trainer output dir (414cf94616fe848b0d773ec954db7fd83723e2e2)


Co-authored-by: Han Chen <hchen725@users.noreply.huggingface.co>

Files changed (1) hide show
  1. geneformer/classifier.py +1 -1
geneformer/classifier.py CHANGED
@@ -1142,7 +1142,7 @@ class Classifier:
1142
  metric="eval_macro_f1",
1143
  metric_columns=["loss", "eval_loss", "eval_accuracy", "eval_macro_f1"],
1144
  ),
1145
- local_dir=output_directory,
1146
  )
1147
 
1148
  return trainer
 
1142
  metric="eval_macro_f1",
1143
  metric_columns=["loss", "eval_loss", "eval_accuracy", "eval_macro_f1"],
1144
  ),
1145
+ storage_path=output_directory,
1146
  )
1147
 
1148
  return trainer