Move the LR scheduler to the last of each epoch

Author: Tony-Y

README.md

@@ -15,7 +15,7 @@ Note that the crystal graph represents only a repeating unit of [a periodic grap
 ## Requirements
 
 * Python 3.7
-* PyTorch 1.0
+* PyTorch 1.1+
 * Pandas
 * Matplotlib (necessary for plotting scripts)
 

src/model_utils.py

@@ -123,8 +123,6 @@ def train(self, train_dl, val_dl, num_epochs):
         for epoch in range(num_epochs):
             epoch_since = time.time()
             print('Epoch {}/{}'.format(epoch, num_epochs - 1), flush=True)
-
-            self.scheduler.step()
             print('current lr:', self.optimizer.param_groups[0]['lr'])
 
             train_val_metrics = []
@@ -148,6 +146,7 @@ def train(self, train_dl, val_dl, num_epochs):
                     checkpoint.check(metric)
                 train_val_metrics += [('_'.join([phase, name]), metric) for name, metric in epoch_metrics]
             history.write(epoch, train_val_metrics)
+            self.scheduler.step()
             time_elapsed = time.time() - epoch_since
             print('Elapsed time (sec.): {:.3f}'.format(time_elapsed))
             print()