(self, input_param, input_shape, expected_shape)
| 55 | |
| 56 | @parameterized.expand([TEST_CASE_0]) |
| 57 | def test_script(self, input_param, input_shape, expected_shape): |
| 58 | spatial_dims = input_param["dim"] |
| 59 | stride = (1, 2, 2)[:spatial_dims] |
| 60 | model = resnet18(spatial_dims=spatial_dims, conv1_t_stride=stride) |
| 61 | input_param["model"] = model |
| 62 | net = NetAdapter(**input_param).to("cpu") |
| 63 | test_data = torch.randn(input_shape).to("cpu") |
| 64 | test_script_save(net, test_data) |
| 65 | |
| 66 | |
| 67 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected