mirror of https://github.com/coqui-ai/TTS.git
bugfix in GST
This commit is contained in:
parent
89d338358e
commit
a2ee48c28e
|
@ -559,8 +559,6 @@ def main(args): # pylint: disable=redefined-outer-name
|
||||||
|
|
||||||
# setup criterion
|
# setup criterion
|
||||||
criterion = TacotronLoss(c, stopnet_pos_weight=10.0, ga_sigma=0.4)
|
criterion = TacotronLoss(c, stopnet_pos_weight=10.0, ga_sigma=0.4)
|
||||||
for name, _ in model.named_parameters():
|
|
||||||
print(name)
|
|
||||||
|
|
||||||
if args.restore_path:
|
if args.restore_path:
|
||||||
checkpoint = torch.load(args.restore_path, map_location='cpu')
|
checkpoint = torch.load(args.restore_path, map_location='cpu')
|
||||||
|
|
Loading…
Reference in New Issue