Purge unused ModelTester code (#37085)
* Purge correctly this time * Remove more methods from recent PRs * make fixup
This commit is contained in:
@@ -75,17 +75,6 @@ class TimmWrapperModelTester:
|
||||
def get_config(self):
|
||||
return TimmWrapperConfig.from_pretrained(self.model_name)
|
||||
|
||||
def create_and_check_model(self, config, pixel_values):
|
||||
model = TimmWrapperModel(config=config)
|
||||
model.to(torch_device)
|
||||
model.eval()
|
||||
with torch.no_grad():
|
||||
result = model(pixel_values)
|
||||
self.parent.assertEqual(
|
||||
result.feature_map[-1].shape,
|
||||
(self.batch_size, model.channels[-1], 14, 14),
|
||||
)
|
||||
|
||||
def prepare_config_and_inputs_for_common(self):
|
||||
config_and_inputs = self.prepare_config_and_inputs()
|
||||
config, pixel_values = config_and_inputs
|
||||
|
||||
Reference in New Issue
Block a user