Skip to content

Commit

Permalink
add blt vs model (#1482)
Browse files Browse the repository at this point in the history
* add blt vs model

* change pretrained import to .pretrained

* change pretrained import to .pretrained in __init__

* remove test.py

* add test

* correct check models import

* improve __init__

* sys path insert for model in init

* sys path insert for pretrained in model.py

* empty test without identifier
  • Loading branch information
kapoorlab authored Nov 26, 2024
1 parent 3147d2c commit 5bacf62
Show file tree
Hide file tree
Showing 5 changed files with 1,218 additions and 0 deletions.
15 changes: 15 additions & 0 deletions brainscore_vision/models/blt-vs/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@

from brainscore_vision import model_registry
from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import sys
import os
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from model import get_model, LAYERS



model_registry['blt_vs'] = lambda: ModelCommitment(
identifier='blt_vs',
activations_model=get_model(),
layers=LAYERS)

Loading

0 comments on commit 5bacf62

Please sign in to comment.