hchen725 commited on
Commit
414cf94
1 Parent(s): 63275a8

Update trainer output dir

Browse files

local_dir deprecated, change to storage_path

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