wasmdashai commited on
Commit
5b25fed
·
verified ·
1 Parent(s): feb3184

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)