rollback resume

This commit is contained in:
Stardust·减
2023-09-04 22:14:46 +08:00
committed by GitHub
parent 7761dadcc6
commit 36b15ba514

View File

@@ -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