Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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():
|