diff --git a/test/expect/ModelTester.test_swin_t_expect.pkl b/test/expect/ModelTester.test_swin_t_expect.pkl index 7326683b7a5..8fb20630cc3 100644 Binary files a/test/expect/ModelTester.test_swin_t_expect.pkl and b/test/expect/ModelTester.test_swin_t_expect.pkl differ diff --git a/test/test_models.py b/test/test_models.py index adbf7e2819e..a3979f4d4e3 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -323,6 +323,7 @@ def _check_input_backprop(model, inputs): "regnet_y_128gf", "regnet_x_16gf", "regnet_x_32gf", + "swin_t", ] for m in slow_models: _model_params[m] = {"input_shape": (1, 3, 64, 64)}