AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[kws] use **kvargs for optim to reduce code (#25)

Browse Source
This commit is contained in:
Binbin Zhang 2021-12-06 11:23:44 +08:00 committed by GitHub
parent 88444ab177
commit a5a54782cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
kws/bin/train.py
View File

@ -195,9 +195,7 @@ def main():
device = torch.device('cuda' if use_cuda else 'cpu') device = torch.device('cuda' if use_cuda else 'cpu')
model = model.to(device) model = model.to(device)
optimizer = optim.Adam(model.parameters(), optimizer = optim.Adam(model.parameters(), **configs['optim_conf'])
lr=configs['optim_conf']['lr'],
weight_decay=configs['optim_conf'].get('weight_decay', 0))
scheduler = optim.lr_scheduler.ReduceLROnPlateau( scheduler = optim.lr_scheduler.ReduceLROnPlateau(
optimizer, optimizer,
mode='min', mode='min',