diff --git a/legged_gym/utils/helpers.py b/legged_gym/utils/helpers.py index dc231af..7dc9716 100644 --- a/legged_gym/utils/helpers.py +++ b/legged_gym/utils/helpers.py @@ -157,8 +157,7 @@ def export_policy_as_jit(actor_critic, path): path = os.path.join(path, 'policy_1.pt') model = copy.deepcopy(actor_critic.actor).to('cpu') traced_script_module = torch.jit.script(model) - traced_script_module.save(path) - + traced_script_module.save(path) class PolicyExporterLSTM(torch.nn.Module): def __init__(self, actor_critic):