Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -124,7 +124,7 @@ training_args = TrainingArguments(
|
|
124 |
#fp16=True,
|
125 |
save_strategy = "no",
|
126 |
load_best_model_at_end=False,
|
127 |
-
push_to_hub=True,
|
128 |
)
|
129 |
print ("training args")
|
130 |
############################################
|
@@ -140,7 +140,7 @@ trainer = Trainer(
|
|
140 |
)
|
141 |
print ("trainer")
|
142 |
#trainer ausführen
|
143 |
-
trainer.train(
|
144 |
#Wenn man vom letzten checkpoint aus weiter trainieren möchte: trainer.train(resume_from_checkpoint=True)
|
145 |
print("trained!!!!!")
|
146 |
#in den Hub laden
|
@@ -179,7 +179,7 @@ print("done")
|
|
179 |
#Push to Hub
|
180 |
print("push to hub")
|
181 |
#login(token=os.environ["HF_ACCESS_TOKEN"])
|
182 |
-
|
183 |
print("done")
|
184 |
|
185 |
|
|
|
124 |
#fp16=True,
|
125 |
save_strategy = "no",
|
126 |
load_best_model_at_end=False,
|
127 |
+
#push_to_hub=True,
|
128 |
)
|
129 |
print ("training args")
|
130 |
############################################
|
|
|
140 |
)
|
141 |
print ("trainer")
|
142 |
#trainer ausführen
|
143 |
+
trainer.train()
|
144 |
#Wenn man vom letzten checkpoint aus weiter trainieren möchte: trainer.train(resume_from_checkpoint=True)
|
145 |
print("trained!!!!!")
|
146 |
#in den Hub laden
|
|
|
179 |
#Push to Hub
|
180 |
print("push to hub")
|
181 |
#login(token=os.environ["HF_ACCESS_TOKEN"])
|
182 |
+
trainer.push_to_hub("alexkueck/test-tis-1", use_auth_token=os.environ["HF_ACCESS_TOKEN"])
|
183 |
print("done")
|
184 |
|
185 |
|