Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ print ("lm datasets")
|
|
104 |
#Training Args
|
105 |
|
106 |
training_args = TrainingArguments(
|
107 |
-
output_dir="alexkueck/
|
108 |
overwrite_output_dir = 'True',
|
109 |
evaluation_strategy = "epoch",
|
110 |
learning_rate=2e-5,
|
@@ -159,7 +159,7 @@ print("Output:\n" )
|
|
159 |
|
160 |
|
161 |
print("Save to Space")
|
162 |
-
trainer.save_model("alexkueck/
|
163 |
print("done")
|
164 |
|
165 |
#####################################
|
|
|
104 |
#Training Args
|
105 |
|
106 |
training_args = TrainingArguments(
|
107 |
+
output_dir="alexkueck/test-tis/",
|
108 |
overwrite_output_dir = 'True',
|
109 |
evaluation_strategy = "epoch",
|
110 |
learning_rate=2e-5,
|
|
|
159 |
|
160 |
|
161 |
print("Save to Space")
|
162 |
+
trainer.save_model("alexkueck/test-tis")
|
163 |
print("done")
|
164 |
|
165 |
#####################################
|