MCPcopy Create free account
hub / github.com/Project-MONAI/MONAI / test_script

Method test_script

tests/networks/nets/test_autoencoder.py:84–87  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

82 self.assertEqual(result.shape, expected_shape)
83
84 def test_script(self):
85 net = AutoEncoder(spatial_dims=2, in_channels=1, out_channels=1, channels=(4, 8), strides=(2, 2))
86 test_data = torch.randn(2, 1, 32, 32)
87 test_script_save(net, test_data)
88
89 def test_channel_stride_difference(self):
90 with self.assertRaises(ValueError):

Callers

nothing calls this directly

Calls 2

AutoEncoderClass · 0.90
test_script_saveFunction · 0.90

Tested by

no test coverage detected