From 36b15ba514a2e2e51471df14fb0c4702780a3deb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Mon, 4 Sep 2023 22:14:46 +0800 Subject: [PATCH] rollback resume --- utils.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/utils.py b/utils.py index bf825ea..0de6fa6 100644 --- a/utils.py +++ b/utils.py @@ -156,7 +156,6 @@ 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) @@ -178,7 +177,6 @@ def get_hparams(init=True): hparams = HParams(**config) hparams.model_dir = model_dir - hparams.resume = args.resume return hparams