update2
This commit is contained in:
@@ -38,6 +38,10 @@ def main():
|
|||||||
with open(config_path, "r", encoding="utf-8") as f:
|
with open(config_path, "r", encoding="utf-8") as f:
|
||||||
cfg = json.load(f)
|
cfg = json.load(f)
|
||||||
|
|
||||||
|
# Resolve config path without duplicating base_dir on Windows when user passes example/config.json
|
||||||
|
if config_path.exists():
|
||||||
|
config_path = resolve_path(config_path.parent, config_path)
|
||||||
|
else:
|
||||||
config_path = resolve_path(base_dir, config_path)
|
config_path = resolve_path(base_dir, config_path)
|
||||||
timesteps = cfg.get("timesteps", 200)
|
timesteps = cfg.get("timesteps", 200)
|
||||||
seq_len = cfg.get("sample_seq_len", cfg.get("seq_len", 64))
|
seq_len = cfg.get("sample_seq_len", cfg.get("seq_len", 64))
|
||||||
|
|||||||
Reference in New Issue
Block a user