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

Update utils.py

Browse Source
This commit is contained in:
Stardust·减
2023-10-04 00:38:36 +08:00
committed by GitHub
parent c03b859beb
commit 068c53bcc6
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
utils.py
View File

@@ -206,15 +206,14 @@ def get_hparams(init=True):
config_path = args.config config_path = args.config
config_save_path = os.path.join(model_dir, "config.json") config_save_path = os.path.join(model_dir, "config.json")
if init: if init:
with open(config_path, "r") as f: with open(config_path, "r", encoding="utf-8") as f:
data = f.read() data = f.read()
with open(config_save_path, "w") as f: with open(config_save_path, "w", encoding="utf-8") as f:
f.write(data) f.write(data)
else: else:
with open(config_save_path, "r") as f: with open(config_save_path, "r", vencoding="utf-8") as f:
data = f.read() data = f.read()
config = json.loads(data) config = json.loads(data)
hparams = HParams(**config) hparams = HParams(**config)
hparams.model_dir = model_dir hparams.model_dir = model_dir
return hparams return hparams