Kevin Fink
commited on
Commit
·
d159076
1
Parent(s):
b6a7390
dev
Browse files
app.py
CHANGED
|
@@ -55,6 +55,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
| 55 |
|
| 56 |
# Set training arguments
|
| 57 |
training_args = TrainingArguments(
|
|
|
|
| 58 |
torch_empty_cache_steps=100,
|
| 59 |
overwrite_output_dir=True,
|
| 60 |
output_dir='/data/results',
|
|
|
|
| 55 |
|
| 56 |
# Set training arguments
|
| 57 |
training_args = TrainingArguments(
|
| 58 |
+
remove_unused_columns=False,
|
| 59 |
torch_empty_cache_steps=100,
|
| 60 |
overwrite_output_dir=True,
|
| 61 |
output_dir='/data/results',
|