diff --git a/test/expect/ModelTester.test_alexnet_expect.pkl b/test/expect/ModelTester.test_alexnet_expect.pkl new file mode 100644 index 00000000000..b21549bec93 Binary files /dev/null and b/test/expect/ModelTester.test_alexnet_expect.pkl differ diff --git a/test/expect/ModelTester.test_densenet121_expect.pkl b/test/expect/ModelTester.test_densenet121_expect.pkl new file mode 100644 index 00000000000..83bcc3282fb Binary files /dev/null and b/test/expect/ModelTester.test_densenet121_expect.pkl differ diff --git a/test/expect/ModelTester.test_densenet161_expect.pkl b/test/expect/ModelTester.test_densenet161_expect.pkl new file mode 100644 index 00000000000..9b221957014 Binary files /dev/null and b/test/expect/ModelTester.test_densenet161_expect.pkl differ diff --git a/test/expect/ModelTester.test_densenet169_expect.pkl b/test/expect/ModelTester.test_densenet169_expect.pkl new file mode 100644 index 00000000000..42743fb0c80 Binary files /dev/null and b/test/expect/ModelTester.test_densenet169_expect.pkl differ diff --git a/test/expect/ModelTester.test_densenet201_expect.pkl b/test/expect/ModelTester.test_densenet201_expect.pkl new file mode 100644 index 00000000000..fd85ca178c9 Binary files /dev/null and b/test/expect/ModelTester.test_densenet201_expect.pkl differ diff --git a/test/expect/ModelTester.test_googlenet_expect.pkl b/test/expect/ModelTester.test_googlenet_expect.pkl new file mode 100644 index 00000000000..2b10ab1f176 Binary files /dev/null and b/test/expect/ModelTester.test_googlenet_expect.pkl differ diff --git a/test/expect/ModelTester.test_inception_v3_expect.pkl b/test/expect/ModelTester.test_inception_v3_expect.pkl new file mode 100644 index 00000000000..38574f498a5 Binary files /dev/null and b/test/expect/ModelTester.test_inception_v3_expect.pkl differ diff --git a/test/expect/ModelTester.test_mnasnet0_5_expect.pkl b/test/expect/ModelTester.test_mnasnet0_5_expect.pkl new file mode 100644 index 00000000000..aaa969d60c0 Binary files /dev/null and b/test/expect/ModelTester.test_mnasnet0_5_expect.pkl differ diff --git a/test/expect/ModelTester.test_mnasnet0_75_expect.pkl b/test/expect/ModelTester.test_mnasnet0_75_expect.pkl new file mode 100644 index 00000000000..44213104b52 Binary files /dev/null and b/test/expect/ModelTester.test_mnasnet0_75_expect.pkl differ diff --git a/test/expect/ModelTester.test_mnasnet1_0_expect.pkl b/test/expect/ModelTester.test_mnasnet1_0_expect.pkl new file mode 100644 index 00000000000..01bc90ba95f Binary files /dev/null and b/test/expect/ModelTester.test_mnasnet1_0_expect.pkl differ diff --git a/test/expect/ModelTester.test_mnasnet1_3_expect.pkl b/test/expect/ModelTester.test_mnasnet1_3_expect.pkl new file mode 100644 index 00000000000..746a5788844 Binary files /dev/null and b/test/expect/ModelTester.test_mnasnet1_3_expect.pkl differ diff --git a/test/expect/ModelTester.test_mobilenet_v2_expect.pkl b/test/expect/ModelTester.test_mobilenet_v2_expect.pkl new file mode 100644 index 00000000000..42bfdc5c68f Binary files /dev/null and b/test/expect/ModelTester.test_mobilenet_v2_expect.pkl differ diff --git a/test/expect/ModelTester.test_mobilenetv2_residual_setting_expect.pkl b/test/expect/ModelTester.test_mobilenetv2_residual_setting_expect.pkl new file mode 100644 index 00000000000..c7733885caf Binary files /dev/null and b/test/expect/ModelTester.test_mobilenetv2_residual_setting_expect.pkl differ diff --git a/test/expect/ModelTester.test_resnet101_expect.pkl b/test/expect/ModelTester.test_resnet101_expect.pkl new file mode 100644 index 00000000000..284aa92b2df Binary files /dev/null and b/test/expect/ModelTester.test_resnet101_expect.pkl differ diff --git a/test/expect/ModelTester.test_resnet152_expect.pkl b/test/expect/ModelTester.test_resnet152_expect.pkl new file mode 100644 index 00000000000..708f554fefa Binary files /dev/null and b/test/expect/ModelTester.test_resnet152_expect.pkl differ diff --git a/test/expect/ModelTester.test_resnet18_expect.pkl b/test/expect/ModelTester.test_resnet18_expect.pkl new file mode 100644 index 00000000000..6dc83a0606f Binary files /dev/null and b/test/expect/ModelTester.test_resnet18_expect.pkl differ diff --git a/test/expect/ModelTester.test_resnet34_expect.pkl b/test/expect/ModelTester.test_resnet34_expect.pkl new file mode 100644 index 00000000000..536075bd40e Binary files /dev/null and b/test/expect/ModelTester.test_resnet34_expect.pkl differ diff --git a/test/expect/ModelTester.test_resnet50_expect.pkl b/test/expect/ModelTester.test_resnet50_expect.pkl new file mode 100644 index 00000000000..0428c5188df Binary files /dev/null and b/test/expect/ModelTester.test_resnet50_expect.pkl differ diff --git a/test/expect/ModelTester.test_resnext101_32x8d_expect.pkl b/test/expect/ModelTester.test_resnext101_32x8d_expect.pkl new file mode 100644 index 00000000000..92061b559ab Binary files /dev/null and b/test/expect/ModelTester.test_resnext101_32x8d_expect.pkl differ diff --git a/test/expect/ModelTester.test_resnext50_32x4d_expect.pkl b/test/expect/ModelTester.test_resnext50_32x4d_expect.pkl new file mode 100644 index 00000000000..4e5dc89c496 Binary files /dev/null and b/test/expect/ModelTester.test_resnext50_32x4d_expect.pkl differ diff --git a/test/expect/ModelTester.test_shufflenet_v2_x0_5_expect.pkl b/test/expect/ModelTester.test_shufflenet_v2_x0_5_expect.pkl new file mode 100644 index 00000000000..72af3d712ed Binary files /dev/null and b/test/expect/ModelTester.test_shufflenet_v2_x0_5_expect.pkl differ diff --git a/test/expect/ModelTester.test_shufflenet_v2_x1_0_expect.pkl b/test/expect/ModelTester.test_shufflenet_v2_x1_0_expect.pkl new file mode 100644 index 00000000000..6feab9857bb Binary files /dev/null and b/test/expect/ModelTester.test_shufflenet_v2_x1_0_expect.pkl differ diff --git a/test/expect/ModelTester.test_shufflenet_v2_x1_5_expect.pkl b/test/expect/ModelTester.test_shufflenet_v2_x1_5_expect.pkl new file mode 100644 index 00000000000..fc36135789f Binary files /dev/null and b/test/expect/ModelTester.test_shufflenet_v2_x1_5_expect.pkl differ diff --git a/test/expect/ModelTester.test_shufflenet_v2_x2_0_expect.pkl b/test/expect/ModelTester.test_shufflenet_v2_x2_0_expect.pkl new file mode 100644 index 00000000000..1af1049efbc Binary files /dev/null and b/test/expect/ModelTester.test_shufflenet_v2_x2_0_expect.pkl differ diff --git a/test/expect/ModelTester.test_squeezenet1_0_expect.pkl b/test/expect/ModelTester.test_squeezenet1_0_expect.pkl new file mode 100644 index 00000000000..12fccaf3f2f Binary files /dev/null and b/test/expect/ModelTester.test_squeezenet1_0_expect.pkl differ diff --git a/test/expect/ModelTester.test_squeezenet1_1_expect.pkl b/test/expect/ModelTester.test_squeezenet1_1_expect.pkl new file mode 100644 index 00000000000..a560181525e Binary files /dev/null and b/test/expect/ModelTester.test_squeezenet1_1_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg11_bn_expect.pkl b/test/expect/ModelTester.test_vgg11_bn_expect.pkl new file mode 100644 index 00000000000..ba3db474ae2 Binary files /dev/null and b/test/expect/ModelTester.test_vgg11_bn_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg11_expect.pkl b/test/expect/ModelTester.test_vgg11_expect.pkl new file mode 100644 index 00000000000..d7a7f3da7e4 Binary files /dev/null and b/test/expect/ModelTester.test_vgg11_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg13_bn_expect.pkl b/test/expect/ModelTester.test_vgg13_bn_expect.pkl new file mode 100644 index 00000000000..e872729eba3 Binary files /dev/null and b/test/expect/ModelTester.test_vgg13_bn_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg13_expect.pkl b/test/expect/ModelTester.test_vgg13_expect.pkl new file mode 100644 index 00000000000..ce387c2ffb7 Binary files /dev/null and b/test/expect/ModelTester.test_vgg13_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg16_bn_expect.pkl b/test/expect/ModelTester.test_vgg16_bn_expect.pkl new file mode 100644 index 00000000000..30cd9c983f6 Binary files /dev/null and b/test/expect/ModelTester.test_vgg16_bn_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg16_expect.pkl b/test/expect/ModelTester.test_vgg16_expect.pkl new file mode 100644 index 00000000000..0e2e4fe0c01 Binary files /dev/null and b/test/expect/ModelTester.test_vgg16_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg19_bn_expect.pkl b/test/expect/ModelTester.test_vgg19_bn_expect.pkl new file mode 100644 index 00000000000..c20b7a22247 Binary files /dev/null and b/test/expect/ModelTester.test_vgg19_bn_expect.pkl differ diff --git a/test/expect/ModelTester.test_vgg19_expect.pkl b/test/expect/ModelTester.test_vgg19_expect.pkl new file mode 100644 index 00000000000..6b42831a1c6 Binary files /dev/null and b/test/expect/ModelTester.test_vgg19_expect.pkl differ diff --git a/test/expect/ModelTester.test_wide_resnet101_2_expect.pkl b/test/expect/ModelTester.test_wide_resnet101_2_expect.pkl new file mode 100644 index 00000000000..cff033c5590 Binary files /dev/null and b/test/expect/ModelTester.test_wide_resnet101_2_expect.pkl differ diff --git a/test/expect/ModelTester.test_wide_resnet50_2_expect.pkl b/test/expect/ModelTester.test_wide_resnet50_2_expect.pkl new file mode 100644 index 00000000000..09bbb39a69c Binary files /dev/null and b/test/expect/ModelTester.test_wide_resnet50_2_expect.pkl differ diff --git a/test/test_models.py b/test/test_models.py index 1864d233772..c70ef6830bf 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -72,11 +72,13 @@ def check_script(self, model, name): def _test_classification_model(self, name, input_shape): # passing num_class equal to a number other than 1000 helps in making the test # more enforcing in nature + set_rng_seed(0) model = models.__dict__[name](num_classes=50) self.check_script(model, name) model.eval() x = torch.rand(input_shape) out = model(x) + self.assertExpected(out, rtol=1e-2, atol=0.) self.assertEqual(out.shape[-1], 50) def _test_segmentation_model(self, name):