wasmdashai commited on
Commit
4986393
·
verified ·
1 Parent(s): b30b753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -598,6 +598,8 @@ def run_train_epoch(num):
598
  for i in range(num):
599
  # model.train(True)
600
  yield pro.run_train_epoch()
 
 
601
 
602
  @spaces.GPU
603
  def init_training():
 
598
  for i in range(num):
599
  # model.train(True)
600
  yield pro.run_train_epoch()
601
+ save_pretrained(pro.dir_model)
602
+ return 'save model '
603
 
604
  @spaces.GPU
605
  def init_training():