Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -105,7 +105,7 @@ print ("lm datasets")
|
|
105 |
batch_size = 2
|
106 |
|
107 |
training_args = TrainingArguments(
|
108 |
-
output_dir="C
|
109 |
overwrite_output_dir = 'True',
|
110 |
per_device_train_batch_size=batch_size, #batch_size = 2 for full training
|
111 |
per_device_eval_batch_size=batch_size,
|
@@ -172,7 +172,7 @@ print("Output:\n" )
|
|
172 |
|
173 |
|
174 |
print("Save to Space")
|
175 |
-
trainer.save_model("C
|
176 |
print("done")
|
177 |
|
178 |
#####################################
|
|
|
105 |
batch_size = 2
|
106 |
|
107 |
training_args = TrainingArguments(
|
108 |
+
output_dir="C:/Users/alexk/Documents/Schule",
|
109 |
overwrite_output_dir = 'True',
|
110 |
per_device_train_batch_size=batch_size, #batch_size = 2 for full training
|
111 |
per_device_eval_batch_size=batch_size,
|
|
|
172 |
|
173 |
|
174 |
print("Save to Space")
|
175 |
+
trainer.save_model("C:/Users/alexk/Documents/Schule")
|
176 |
print("done")
|
177 |
|
178 |
#####################################
|