Lord-Raven commited on
Commit
4d2ee65
·
1 Parent(s): 74f03e3

Experimenting with few-shot classification.

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -75,6 +75,7 @@ args = TrainingArguments(
75
 
76
  trainer = Trainer(
77
  model=few_shot_model,
 
78
  train_dataset=train_dataset,
79
  eval_dataset=reference_dataset["test"]
80
  )
 
75
 
76
  trainer = Trainer(
77
  model=few_shot_model,
78
+ args=args,
79
  train_dataset=train_dataset,
80
  eval_dataset=reference_dataset["test"]
81
  )