Spaces:
Sleeping
Sleeping
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)
|