Spaces:
Running
Running
update
Browse files
examples/mpnet_aishell/step_2_train_model.py
CHANGED
@@ -230,7 +230,7 @@ def main():
|
|
230 |
patience_count = 0
|
231 |
|
232 |
logger.info("training")
|
233 |
-
for idx_epoch in range(max(0, last_epoch), args.max_epochs):
|
234 |
# train
|
235 |
generator.train()
|
236 |
discriminator.train()
|
|
|
230 |
patience_count = 0
|
231 |
|
232 |
logger.info("training")
|
233 |
+
for idx_epoch in range(max(0, last_epoch+1), args.max_epochs):
|
234 |
# train
|
235 |
generator.train()
|
236 |
discriminator.train()
|