Always use training loss for lr scheduler

2020-07-27 17:47:58 -07:00 · 2020-07-27 17:47:58 -07:00 · a886e53c54
commit a886e53c54
parent 2336d83f2d
1 changed files with 1 additions and 1 deletions
--- a/map2map/train.py
+++ b/map2map/train.py
@ -207,7 +207,7 @@ def gpu_worker(local_rank, node, args):
        if args.val:
            val_loss = validate(epoch, val_loader, model, lag2eul, criterion,
                logger, device, args)
-            epoch_loss = val_loss
+            #epoch_loss = val_loss

        if args.reduce_lr_on_plateau:
            lag_scheduler.step(epoch_loss[0])