Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

resolve optimizer restore problem

This commit is contained in:
Eren Golge 2018-04-28 13:27:56 -07:00
parent 5f7bf85b19
commit 0ac6284655
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
train.py
View File

@ -351,8 +351,6 @@ def main(args):
c.num_mels, c.num_mels,
c.r) c.r)
optimizer = optim.Adam(model.parameters(), lr=c.lr)
if use_cuda: if use_cuda:
criterion = L1LossMasked().cuda() criterion = L1LossMasked().cuda()
else: else:
@ -361,7 +359,12 @@ def main(args):
if args.restore_path: if args.restore_path:
checkpoint = torch.load(args.restore_path) checkpoint = torch.load(args.restore_path)
model.load_state_dict(checkpoint['model']) model.load_state_dict(checkpoint['model'])
optimizer = optim.Adam(model.parameters(), lr=c.lr)
optimizer.load_state_dict(checkpoint['optimizer']) optimizer.load_state_dict(checkpoint['optimizer'])
for state in optimizer.state.values():
for k, v in state.items():
if torch.is_tensor(v):
state[k] = v.cuda()
print(" > Model restored from step %d" % checkpoint['step']) print(" > Model restored from step %d" % checkpoint['step'])
start_epoch = checkpoint['step'] // len(train_loader) start_epoch = checkpoint['step'] // len(train_loader)
best_loss = checkpoint['linear_loss'] best_loss = checkpoint['linear_loss']
@ -369,6 +372,7 @@ def main(args):
args.restore_step = checkpoint['step'] args.restore_step = checkpoint['step']
else: else:
args.restore_step = 0 args.restore_step = 0
optimizer = optim.Adam(model.parameters(), lr=c.lr)
print(" > Starting a new training") print(" > Starting a new training")
if use_cuda: if use_cuda: