From 28b59cebb1d9929ba177ff51ce19193af76f92fc Mon Sep 17 00:00:00 2001 From: Joao Gomes Date: Wed, 27 Apr 2022 15:04:24 +0100 Subject: [PATCH 1/3] Add swin_t to slow list --- test/test_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/test_models.py b/test/test_models.py index adbf7e2819e..d8f90d55a55 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)} From 1a35cc917f52fbd9c57848c32247c33e3cab5781 Mon Sep 17 00:00:00 2001 From: Joao Gomes Date: Wed, 27 Apr 2022 18:19:47 +0100 Subject: [PATCH 2/3] apply ufmt --- test/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_models.py b/test/test_models.py index d8f90d55a55..a3979f4d4e3 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -323,7 +323,7 @@ def _check_input_backprop(model, inputs): "regnet_y_128gf", "regnet_x_16gf", "regnet_x_32gf", - "swin_t" + "swin_t", ] for m in slow_models: _model_params[m] = {"input_shape": (1, 3, 64, 64)} From 1a3d20aeb165bb0582e9b7716bb51669f01aebc2 Mon Sep 17 00:00:00 2001 From: Joao Gomes Date: Wed, 27 Apr 2022 18:26:35 +0100 Subject: [PATCH 3/3] update expect for swin_t --- test/expect/ModelTester.test_swin_t_expect.pkl | Bin 939 -> 939 bytes 1 file changed, 0 insertions(+), 0 deletions(-) diff --git a/test/expect/ModelTester.test_swin_t_expect.pkl b/test/expect/ModelTester.test_swin_t_expect.pkl index 7326683b7a5adbc48f0f84087f5a4de56f585d64..8fb20630cc363e5b5297d35d29eeeb15756c10c7 100644 GIT binary patch delta 230 zcmV${#<4f``3eqei}hI~70Pb(B8A$OOOlYo5NM$CW~m#sh6a;m;)nF~Lt4GBL!RqnpV9&J&+#*ojxgwf(YL&Wc$y)*g5yT@ErNQwOFIR6kVFWaLaSPd^HL`Jc7f%t??tjfbgI_MqVzz g5Ciu-;7n`2P)i30usa|4lMn*X1h6|F_><%U$F)gsX8-^I delta 230 zcmV*|FW1GHHd