mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Update VitDetModelTester.get_config
to use pretrain_image_size
(#27831)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
parent
28e2887a1a
commit
ac975074e6
@ -90,6 +90,7 @@ class VitDetModelTester:
|
||||
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,
|
||||
|
Loading…
Reference in New Issue
Block a user