EmbodiedAI
/
walk-these-ways-go2
mirror of https://github.com/Teddy-Liao/walk-these-ways-go2.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update deploy_policy.py

This commit is contained in:
Liao Dengting 2024-01-29 10:06:53 +08:00
parent f416bee1f7
commit a5e2eb25e3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
go2_gym_deploy/scripts/deploy_policy.py
View File

@ -18,8 +18,8 @@ def load_and_run_policy(label, experiment_name, max_vel=1.0, max_yaw_vel=1.0):
logdir = sorted(dirs)[0] logdir = sorted(dirs)[0]
# with open(logdir+"/parameters.pkl", 'rb') as file: # with open(logdir+"/parameters.pkl", 'rb') as file:
with open(logdir+"/parameters_cpu.pkl", 'rb') as file: with open(logdir+"/parameters.pkl", 'rb') as file:
pkl_cfg = pkl.load(file).to('cpu') pkl_cfg = pkl.load(file)
print(pkl_cfg.keys()) print(pkl_cfg.keys())
cfg = pkl_cfg["Cfg"] cfg = pkl_cfg["Cfg"]
print(cfg.keys()) print(cfg.keys())