tsai

tsai copied to clipboard

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