Skip to content

Commit

Permalink
Update VitDetModelTester.get_config to use pretrain_image_size (#…
Browse files Browse the repository at this point in the history
…27831)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
  • Loading branch information
ydshieh and ydshieh authored Dec 5, 2023
1 parent 28e2887 commit ac97507
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/models/vitdet/test_modeling_vitdet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit ac97507

Please sign in to comment.