BC算法实现
This commit is contained in:
17
visualize_bc.py
Normal file
17
visualize_bc.py
Normal 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()
|
||||
Reference in New Issue
Block a user