Update helpers.py

This commit is contained in:
Ziluo Ding 2024-07-01 16:26:40 +08:00
parent e9107dfb7c
commit 6f298b4435

View File

@ -159,7 +159,6 @@ def export_policy_as_jit(actor_critic, path):
traced_script_module = torch.jit.script(model)
traced_script_module.save(path)
class PolicyExporterLSTM(torch.nn.Module):
def __init__(self, actor_critic):
super().__init__()