holy fuking shit

2026-09-12 01:56:34 +08:00
parent 36f354736c
commit d975309a81
83 changed files with 2748 additions and 896 deletions
@@ -130,7 +130,8 @@ def main(argv=None) -> int:
         for rank in range(max(1, args.envs))])
    if args.init:
        metadata = json.loads(Path(args.init).with_suffix('.json').read_text(encoding='utf-8'))
-        if metadata.get('algorithm', 'ppo') != args.algorithm or metadata.get('feature_version') != 2:
+        if (metadata.get('algorithm', 'ppo') != args.algorithm
+                or metadata.get('feature_version') != 2):
            parser.error('初始化模型的算法/特征版本与当前训练不符，请重训或提供v2初始化模型')
        model = model_cls.load(args.init, env=vec_env, device=device)
    else: