Merge pull request #7 from tomrunia/regression

minor clean

Author: tomrunia

train.py

@@ -115,13 +115,6 @@
     milestones = [int(x) for x in config.lr_scheduler_milestones.split(',')]
     scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones, config.lr_scheduler_gamma)
 
-####################################################################
-####################################################################
-# Resume training from previous checkpoint
-
-# if config.checkpoint_file:
-#     model_factory.model_restore_checkpoint(config, model)
-
 ####################################################################
 ####################################################################