add --resume option

utils.py

@@ -158,6 +158,7 @@ def get_hparams(init=True):
                         help='JSON file for configuration')
     parser.add_argument('-m', '--model', type=str, required=True,
                         help='Model name')
+    parser.add_argument('--resume', dest='resume', action="store_true", default=False, help="resume training from latest checkpoint of the given model")
 
     args = parser.parse_args()
     model_dir = os.path.join("./logs", args.model)
@@ -179,6 +180,7 @@ def get_hparams(init=True):
 
     hparams = HParams(**config)
     hparams.model_dir = model_dir
+    hparams.resume = args.resume
     return hparams