Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -437,10 +437,10 @@ class TrinerModelVITS:
|
|
437 |
# eps=training_args.adam_epsilon,
|
438 |
)
|
439 |
lr_scheduler = torch.optim.lr_scheduler.ExponentialLR(
|
440 |
-
optimizer, last_epoch=-1
|
441 |
)
|
442 |
disc_lr_scheduler = torch.optim.lr_scheduler.ExponentialLR(
|
443 |
-
disc_optimizer, last_epoch=-1
|
444 |
)
|
445 |
self.models=(self.model,discriminator)
|
446 |
self.optimizers=(optimizer,disc_optimizer,scaler)
|
|
|
437 |
# eps=training_args.adam_epsilon,
|
438 |
)
|
439 |
lr_scheduler = torch.optim.lr_scheduler.ExponentialLR(
|
440 |
+
optimizer,gamma=0.999875, last_epoch=-1
|
441 |
)
|
442 |
disc_lr_scheduler = torch.optim.lr_scheduler.ExponentialLR(
|
443 |
+
disc_optimizer, gamma=0.999875,last_epoch=-1
|
444 |
)
|
445 |
self.models=(self.model,discriminator)
|
446 |
self.optimizers=(optimizer,disc_optimizer,scaler)
|