PR comments

brain-score · May 16, 2024 · 9fb6a02 · 9fb6a02
1 parent decada2
commit 9fb6a02
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 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
@@ -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
@@ -1,5 +1,4 @@
 numpy
 torch
 transformers
-ssl
 PIL