From a5e2eb25e306bd7b4939c9ac95872e6e3fe62727 Mon Sep 17 00:00:00 2001 From: Liao Dengting <540614105@qq.com> Date: Mon, 29 Jan 2024 10:06:53 +0800 Subject: [PATCH] Update deploy_policy.py --- go2_gym_deploy/scripts/deploy_policy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/go2_gym_deploy/scripts/deploy_policy.py b/go2_gym_deploy/scripts/deploy_policy.py index e58e6bb..5f62432 100644 --- a/go2_gym_deploy/scripts/deploy_policy.py +++ b/go2_gym_deploy/scripts/deploy_policy.py @@ -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] # with open(logdir+"/parameters.pkl", 'rb') as file: - with open(logdir+"/parameters_cpu.pkl", 'rb') as file: - pkl_cfg = pkl.load(file).to('cpu') + with open(logdir+"/parameters.pkl", 'rb') as file: + pkl_cfg = pkl.load(file) print(pkl_cfg.keys()) cfg = pkl_cfg["Cfg"] print(cfg.keys())