when LSTM model save as torchScript ( torch.jit.script(model)) it except error

[MichaelCarrik]

torch.jit.script(model).save("model_test_script.pt")
^^^^^^^^^^^^^^^^^^^^^^^

File "/usr/local/lib64/python3.11/site-packages/torch/jit/_script.py", line 1429, in script
ret = _script_impl(
^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_script.py", line 1147, in _script_impl
return torch.jit._recursive.create_script_module(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 557, in create_script_module
return create_script_module_impl(nn_module, concrete_type, stubs_fn)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 630, in create_script_module_impl
script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_script.py", line 650, in _construct
init_fn(script_module)
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 606, in init_fn
scripted = create_script_module_impl(
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 630, in create_script_module_impl
script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_script.py", line 650, in _construct
init_fn(script_module)
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 606, in init_fn
scripted = create_script_module_impl(
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 630, in create_script_module_impl
script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_script.py", line 650, in _construct
init_fn(script_module)
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 606, in init_fn
scripted = create_script_module_impl(
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 630, in create_script_module_impl
script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_script.py", line 650, in _construct
init_fn(script_module)
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 606, in init_fn
scripted = create_script_module_impl(
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 634, in create_script_module_impl
create_methods_and_properties_from_stubs(
File "/usr/local/lib64/python3.11/site-packages/torch/jit/_recursive.py", line 466, in create_methods_and_properties_from_stubs
concrete_type._create_methods_and_properties(
RuntimeError:
Could not cast value of type Tuple[()] to bool:
File "/usr/local/lib/python3.11/site-packages/tsai/models/layers.py", line 429
def forward(self, x):
return x.contiguous().reshape(x.shape[0], -1) if not self.shape else x.contiguous().reshape(-1) if self.shape == (-1,) else x.contiguous().reshape(x.shape[0], *self.shape)
~~~~~~~~~~ <--- HERE