Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -598,7 +598,9 @@ def run_train_epoch(num):
|
|
| 598 |
for i in range(num):
|
| 599 |
# model.train(True)
|
| 600 |
yield pro.run_train_epoch()
|
| 601 |
-
|
|
|
|
|
|
|
| 602 |
return 'save model '
|
| 603 |
|
| 604 |
@spaces.GPU
|
|
|
|
| 598 |
for i in range(num):
|
| 599 |
# model.train(True)
|
| 600 |
yield pro.run_train_epoch()
|
| 601 |
+
|
| 602 |
+
pro.save_pretrained(pro.dir_model)
|
| 603 |
+
pro.load_model()
|
| 604 |
return 'save model '
|
| 605 |
|
| 606 |
@spaces.GPU
|