From 9fb6a023a335f16014251ee66cc297a6aa7bfce1 Mon Sep 17 00:00:00 2001 From: Mike Ferguson Date: Thu, 16 May 2024 11:43:28 -0400 Subject: [PATCH] PR comments --- .../cvt_cvt-w24-384-in22k_finetuned-in1k_4/model.py | 13 +++++++++++-- .../requirements.txt | 1 - 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/model.py b/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/model.py index 1f64a84a1..900f6573f 100644 --- a/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/model.py +++ b/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/model.py @@ -1,10 +1,11 @@ from brainscore_vision.model_helpers.check_submission import check_models import functools +import numpy as np +import torch from transformers import AutoFeatureExtractor, CvtForImageClassification from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper from PIL import Image -import numpy as np -import torch + """ Template module for a base model submission to brain-score @@ -143,6 +144,14 @@ def create_static_video(image, num_frames, normalize_0to1=False, channel_dim=3): # It executes a mock run of brain-score benchmarks. check_models.check_base_models(__name__) + +''' +Below Notes are from the original model file from Brain-Score 1.0, and +kept in this file for posterity. + +''' + + """ Notes on the error: diff --git a/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/requirements.txt b/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/requirements.txt index aaf762ea5..304640474 100644 --- a/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/requirements.txt +++ b/brainscore_vision/models/cvt_cvt-w24-384-in22k_finetuned-in1k_4/requirements.txt @@ -1,5 +1,4 @@ numpy torch transformers -ssl PIL \ No newline at end of file