(self)
| 778 | class TorchImageTestCase3D(NumpyImageTestCase3D): |
| 779 | |
| 780 | def setUp(self): |
| 781 | NumpyImageTestCase3D.setUp(self) |
| 782 | self.imt = torch.tensor(self.imt) |
| 783 | self.seg1 = torch.tensor(self.seg1) |
| 784 | self.segn = torch.tensor(self.segn) |
| 785 | |
| 786 | |
| 787 | def test_script_save(net, *inputs, device=None, rtol=1e-4, atol=0.0): |