Nofated095
/
so-vits-svc
mirror of https://github.com/svc-develop-team/so-vits-svc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Debug

This commit is contained in:
ylzz1997 2023-05-19 18:45:30 +08:00
parent 9c341c13cc
commit f5c3909289
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_diff.py
View File

@ -48,9 +48,9 @@ if __name__ == '__main__':
optimizer = torch.optim.AdamW(model.parameters())
initial_global_step, model, optimizer = utils.load_model(args.env.expdir, model, optimizer, device=args.device)
for param_group in optimizer.param_groups:
param_group['lr'] = args.train.lr
param_group['lr'] = args.train.lr * (args.train.gamma ** (initial_global_step//args.train.decay_step) )
param_group['weight_decay'] = args.train.weight_decay
scheduler = lr_scheduler.StepLR(optimizer, step_size=args.train.decay_step, gamma=args.train.gamma)
scheduler = lr_scheduler.StepLR(optimizer, step_size=args.train.decay_step, gamma=args.train.gamma,last_epoch=initial_global_step-1)
# device
if args.device == 'cuda':