Closed
Description
[2024-07-20T23:15:47.067Z] ======================================================================
[2024-07-20T23:15:47.067Z] ERROR: test_script_00 (tests.test_vit.TestViT)
[2024-07-20T23:15:47.067Z] ----------------------------------------------------------------------
[2024-07-20T23:15:47.067Z] Traceback (most recent call last):
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/parameterized/parameterized.py", line 620, in standalone_func
[2024-07-20T23:15:47.067Z] return func(*(a + p.args), **p.kwargs, **kw)
[2024-07-20T23:15:47.067Z] File "/home/jenkins/agent/workspace/Monai-latest-image/tests/test_vit.py", line 114, in test_script
[2024-07-20T23:15:47.067Z] torch.jit.script(net)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_script.py", line 1338, in script
[2024-07-20T23:15:47.067Z] return torch.jit._recursive.create_script_module(
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 558, in create_script_module
[2024-07-20T23:15:47.067Z] return create_script_module_impl(nn_module, concrete_type, stubs_fn)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 631, in create_script_module_impl
[2024-07-20T23:15:47.067Z] script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_script.py", line 647, in _construct
[2024-07-20T23:15:47.067Z] init_fn(script_module)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 607, in init_fn
[2024-07-20T23:15:47.067Z] scripted = create_script_module_impl(
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 631, in create_script_module_impl
[2024-07-20T23:15:47.067Z] script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_script.py", line 647, in _construct
[2024-07-20T23:15:47.067Z] init_fn(script_module)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 607, in init_fn
[2024-07-20T23:15:47.067Z] scripted = create_script_module_impl(
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 631, in create_script_module_impl
[2024-07-20T23:15:47.067Z] script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_script.py", line 647, in _construct
[2024-07-20T23:15:47.067Z] init_fn(script_module)
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 607, in init_fn
[2024-07-20T23:15:47.067Z] scripted = create_script_module_impl(
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 635, in create_script_module_impl
[2024-07-20T23:15:47.067Z] create_methods_and_properties_from_stubs(
[2024-07-20T23:15:47.067Z] File "/usr/local/lib/python3.10/dist-packages/torch/jit/_recursive.py", line 467, in create_methods_and_properties_from_stubs
[2024-07-20T23:15:47.067Z] concrete_type._create_methods_and_properties(
[2024-07-20T23:15:47.067Z] RuntimeError:
[2024-07-20T23:15:47.067Z] Module 'SABlock' has no attribute 'causal_mask' :
[2024-07-20T23:15:47.067Z] File "/home/jenkins/agent/workspace/Monai-latest-image/monai/networks/blocks/selfattention.py", line 137
[2024-07-20T23:15:47.067Z]
[2024-07-20T23:15:47.067Z] if self.causal:
[2024-07-20T23:15:47.067Z] att_mat = att_mat.masked_fill(self.causal_mask[:, :, : x.shape[1], : x.shape[1]] == 0, float("-inf"))
[2024-07-20T23:15:47.067Z] ~~~~~~~~~~~~~~~~ <--- HERE
[2024-07-20T23:15:47.067Z]
[2024-07-20T23:15:47.067Z] att_mat = att_mat.softmax(dim=-1)
[2024-07-20T23:15:47.067Z]