Update main.py (pytorch#65)

WuCPMark · Jan 30, 2017 · 832141b · 832141b
1 parent b59153a
commit 832141b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imagenet/main.py b/imagenet/main.py
@@ -263,7 +263,7 @@ def update(self, val, n=1):
 def adjust_learning_rate(optimizer, epoch):
     """Sets the learning rate to the initial LR decayed by 10 every 30 epochs"""
     lr = args.lr * (0.1 ** (epoch // 30))
-    for param_group in optimizer.state_dict()['param_groups']:
+    for param_group in optimizer.param_groups:
         param_group['lr'] = lr