Update app.py

app.py

@@ -104,12 +104,23 @@ print ("lm datasets")
 #Training Args
  
 training_args = TrainingArguments(
-    output_dir="./test-tis/",
+    output_dir="./",
     overwrite_output_dir = 'True',
-    evaluation_strategy = "epoch",
+    per_device_train_batch_size=batch_size,  #batch_size = 2 for full training
+    per_device_eval_batch_size=batch_size,
+    evaluation_strategy = "epoch",  #oder steps
     learning_rate=2e-5,
     weight_decay=0.01,
     save_total_limit = 2,
+    predict_with_generate=True,
+    #logging_steps=2,  # set to 1000 for full training
+    #save_steps=16,    # set to 500 for full training
+    #eval_steps=4,     # set to 8000 for full training
+    #warmup_steps=1,   # set to 2000 for full training
+    #max_steps=16,     # delete for full training
+    # overwrite_output_dir=True,
+    #save_total_limit=1,
+    #fp16=True,
     save_strategy = "no",
     load_best_model_at_end=False,
     #push_to_hub=True,
@@ -129,6 +140,7 @@ trainer = Trainer(
 print ("trainer")
 #trainer ausführen
 trainer.train()
+#Wenn man vom letzten checkpoint aus weiter trainieren möchte: trainer.train(resume_from_checkpoint=True)
 print("trained!!!!!")
 #in den Hub laden
 #trainer.push_to_hub("test-tis", use_auth_token=True)
@@ -159,7 +171,7 @@ print("Output:\n" )
 
 
 print("Save to Space")
-trainer.save_model("./test-tis")
+trainer.save_model("./")
 print("done")
 
 #####################################