tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

rollback resume

Browse Source
This commit is contained in:
Stardust·减
2023-09-04 22:14:46 +08:00
committed by GitHub
parent 7761dadcc6
commit 36b15ba514
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils.py
View File

@@ -156,7 +156,6 @@ def get_hparams(init=True):
help='JSON file for configuration') help='JSON file for configuration')
parser.add_argument('-m', '--model', type=str, required=True, parser.add_argument('-m', '--model', type=str, required=True,
help='Model name') 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() args = parser.parse_args()
model_dir = os.path.join("./logs", args.model) model_dir = os.path.join("./logs", args.model)
@@ -178,7 +177,6 @@ def get_hparams(init=True):
hparams = HParams(**config) hparams = HParams(**config)
hparams.model_dir = model_dir hparams.model_dir = model_dir
hparams.resume = args.resume
return hparams return hparams