BC算法实现

This commit is contained in:
2026-02-02 01:18:18 +08:00
parent 265b0eade1
commit 21c046aef0
25 changed files with 632 additions and 322 deletions

17
visualize_bc.py Normal file
View File

@@ -0,0 +1,17 @@
"""
Thin wrapper: forwards to scripts/visualize_trained_policy.py --policy_type bc.
Use: python visualize_bc.py [--model_path models/bc/policy_best.pt] [other args...]
Or call directly: python scripts/visualize_trained_policy.py --policy_type bc --model_path models/bc/policy_best.pt
"""
import subprocess
import sys
import os
def main():
script_dir = os.path.dirname(os.path.abspath(__file__))
script = os.path.join(script_dir, "scripts", "visualize_trained_policy.py")
cmd = [sys.executable, script, "--policy_type", "bc"] + sys.argv[1:]
sys.exit(subprocess.run(cmd).returncode)
if __name__ == "__main__":
main()