Spaces:

wasmdashai
/

RunTasking

Sleeping

wasmdashai commited on Aug 24, 2024

Commit

b050b2e

verified ·

1 Parent(s): b60bf0e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -643,7 +643,7 @@ def run_train_epoch(num):
      pro.init_training()
      for i in range(num):
     # model.train(True)
-        yield pro.run_train_epoch()
    else:
       pro.save_pretrained(pro.dir_model)
       pro.load_model()

      pro.init_training()
      for i in range(num):
     # model.train(True)
+        return  pro.run_train_epoch()
    else:
       pro.save_pretrained(pro.dir_model)
       pro.load_model()