From b04ee340356f044f695525f099542d560c821064 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 4 Sep 2018 14:41:52 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 514195c2..626bbcc0 100644 --- a/train.py +++ b/train.py @@ -68,7 +68,7 @@ def main(opt): # optimizer = torch.optim.SGD(model.parameters(), lr=.001, momentum=.9, weight_decay=5e-4, nesterov=True) # optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters())) optimizer = torch.optim.Adam(model.parameters()) - #optimizer.load_state_dict(checkpoint['optimizer']) + optimizer.load_state_dict(checkpoint['optimizer']) start_epoch = checkpoint['epoch'] + 1 best_loss = checkpoint['best_loss']