(self)
| 57 | |
| 58 | @TimedCall(seconds=800, force_quit=True) |
| 59 | def test_script(self): |
| 60 | input_param, input_shape, expected_shape = TEST_CASE_1 |
| 61 | net = HighResNet(**input_param) |
| 62 | test_data = torch.randn(input_shape) |
| 63 | test_script_save(net, test_data, rtol=1e-4, atol=1e-4) |
| 64 | |
| 65 | |
| 66 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected