wasmdashai commited on
Commit
25942bd
·
verified ·
1 Parent(s): d5e8ac8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -410,14 +410,14 @@ class TrinerModelVITS:
410
  # self.training_args=load_training_args(self.path_training_args)
411
  # training_args= self.training_args
412
  # scaler = GradScaler(enabled=True)
413
- for disc in self.model.discriminator.discriminators:
414
- disc.apply_weight_norm()
415
- self.model.decoder.apply_weight_norm()
416
- # torch.nn.utils.weight_norm(self.decoder.conv_pre)
417
- # torch.nn.utils.weight_norm(self.decoder.conv_post)
418
- for flow in self.model.flow.flows:
419
- torch.nn.utils.weight_norm(flow.conv_pre)
420
- torch.nn.utils.weight_norm(flow.conv_post)
421
 
422
  discriminator = self.model.discriminator
423
  self.model.discriminator = None
 
410
  # self.training_args=load_training_args(self.path_training_args)
411
  # training_args= self.training_args
412
  # scaler = GradScaler(enabled=True)
413
+ # for disc in self.model.discriminator.discriminators:
414
+ # disc.apply_weight_norm()
415
+ # self.model.decoder.apply_weight_norm()
416
+ # # torch.nn.utils.weight_norm(self.decoder.conv_pre)
417
+ # # torch.nn.utils.weight_norm(self.decoder.conv_post)
418
+ # for flow in self.model.flow.flows:
419
+ # torch.nn.utils.weight_norm(flow.conv_pre)
420
+ # torch.nn.utils.weight_norm(flow.conv_post)
421
 
422
  discriminator = self.model.discriminator
423
  self.model.discriminator = None