Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Always use training loss for lr scheduler

Browse Source
This commit is contained in:
Yin Li 2020-07-27 17:47:58 -07:00
parent 2336d83f2d
commit a886e53c54
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/train.py
View File

@ -207,7 +207,7 @@ def gpu_worker(local_rank, node, args):
if args.val: if args.val:
val_loss = validate(epoch, val_loader, model, lag2eul, criterion, val_loss = validate(epoch, val_loader, model, lag2eul, criterion,
logger, device, args) logger, device, args)
epoch_loss = val_loss #epoch_loss = val_loss
if args.reduce_lr_on_plateau: if args.reduce_lr_on_plateau:
lag_scheduler.step(epoch_loss[0]) lag_scheduler.step(epoch_loss[0])