From 47106960dfb5c22918205944e9438093d7399d27 Mon Sep 17 00:00:00 2001 From: AlvinC Date: Mon, 12 May 2025 18:59:36 +0800 Subject: [PATCH] Update play.py to adapt breaking change from Isaaclab v2.1.0 that depends on rsl-rl-lib v2.3.1 --- scripts/rsl_rl/play.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/rsl_rl/play.py b/scripts/rsl_rl/play.py index e381348..dde86d4 100644 --- a/scripts/rsl_rl/play.py +++ b/scripts/rsl_rl/play.py @@ -95,10 +95,10 @@ def main(): # export policy to onnx/jit export_model_dir = os.path.join(os.path.dirname(resume_path), "exported") export_policy_as_jit( - ppo_runner.alg.actor_critic, ppo_runner.obs_normalizer, path=export_model_dir, filename="policy.pt" + ppo_runner.alg.policy, ppo_runner.obs_normalizer, path=export_model_dir, filename="policy.pt" ) export_policy_as_onnx( - ppo_runner.alg.actor_critic, normalizer=ppo_runner.obs_normalizer, path=export_model_dir, filename="policy.onnx" + ppo_runner.alg.policy, normalizer=ppo_runner.obs_normalizer, path=export_model_dir, filename="policy.onnx" ) # reset environment