Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
4d2ee65
1
Parent(s):
74f03e3
Experimenting with few-shot classification.
Browse files
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 |
)
|