Skip to content

Commit

Permalink
Update model.py
Browse files Browse the repository at this point in the history
  • Loading branch information
pellegreene authored Jul 30, 2024
1 parent 87239e9 commit 40bef92
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions brainscore_vision/models/inception_v4_pytorch/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,11 +21,6 @@

MODEL = timm.create_model('inception_v4', pretrained=True)


def get_model_list():
return ['inception_v4_pytorch']


def get_model(name):
assert name == 'inception_v4_pytorch'
preprocessing = functools.partial(load_preprocess_images, image_size=299)
Expand Down Expand Up @@ -62,4 +57,4 @@ def get_bibtex(model_identifier):


if __name__ == '__main__':
check_models.check_base_models(__name__)
check_models.check_base_models(__name__)

0 comments on commit 40bef92

Please sign in to comment.