alexkueck commited on
Commit
d1ec3f8
·
1 Parent(s): 7f61353

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -110,7 +110,7 @@ training_args = TrainingArguments(
110
  evaluation_strategy = "epoch",
111
  learning_rate=2e-5,
112
  weight_decay=0.01,
113
- push_to_hub=True,
114
  )
115
  print ("training args")
116
  ############################################
@@ -130,7 +130,7 @@ trainer.train()
130
  #in den Hub laden
131
  #trainer.push_to_hub("test-tis", use_auth_token=True)
132
  #trainer.push_to_hub()
133
- trainer.save_model("model", use_auth_token=True)
134
 
135
 
136
 
 
110
  evaluation_strategy = "epoch",
111
  learning_rate=2e-5,
112
  weight_decay=0.01,
113
+ #push_to_hub=True,
114
  )
115
  print ("training args")
116
  ############################################
 
130
  #in den Hub laden
131
  #trainer.push_to_hub("test-tis", use_auth_token=True)
132
  #trainer.push_to_hub()
133
+ trainer.save_model("model")
134
 
135
 
136