Update helpers.py
This commit is contained in:
parent
e9107dfb7c
commit
6f298b4435
@ -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__()
|
||||
|
Loading…
x
Reference in New Issue
Block a user