diff --git a/utils.py b/utils.py index 1798c77..56c4955 100644 --- a/utils.py +++ b/utils.py @@ -149,7 +149,7 @@ def get_hparams(init=True): help='Model name') parser.add_argument('-p', '--modeldir', type=str, default="default", - help='Checkpoint save path') + help='Directory to save models') args = parser.parse_args() if args.modeldir == "default":