From ac975074e69eccfb4cea3433f2041029d0f7be9f Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 5 Dec 2023 16:33:27 +0100 Subject: [PATCH] Update `VitDetModelTester.get_config` to use `pretrain_image_size` (#27831) fix Co-authored-by: ydshieh --- tests/models/vitdet/test_modeling_vitdet.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/vitdet/test_modeling_vitdet.py b/tests/models/vitdet/test_modeling_vitdet.py index a18fdda96a5d90..45ee0acee32155 100644 --- a/tests/models/vitdet/test_modeling_vitdet.py +++ b/tests/models/vitdet/test_modeling_vitdet.py @@ -90,6 +90,7 @@ def prepare_config_and_inputs(self): def get_config(self): return VitDetConfig( image_size=self.image_size, + pretrain_image_size=self.image_size, patch_size=self.patch_size, num_channels=self.num_channels, hidden_size=self.hidden_size,