alexkueck commited on
Commit
a6ca5ee
·
1 Parent(s): f88ee53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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(use_auth_token=os.environ["HF_ACCESS_TOKEN"])
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
- #trainer.push_to_hub("alexkueck/test-tis-1")
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