Skip to content

Commit

Permalink
fix version
Browse files Browse the repository at this point in the history
  • Loading branch information
ben22020 committed Jan 28, 2025
1 parent b82ed44 commit 0e3832b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_01/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def get_model(name):

model = alexnet_v2_pytorch()
# Load the pretrained weights
weights_path = load_weight_file(bucket = "brainscorevariability", relative_path = "model_weights/training_seed_01.pth", version_id = "null", sha1 = "4b1bb7810d5288631c04cf4cde882540d6ebee77" ) #new hash
weights_path = load_weight_file(bucket = "brainscorevariability", relative_path = "model_weights/training_seed_01.pth", version_id = None, sha1 = "4b1bb7810d5288631c04cf4cde882540d6ebee77" ) #new hash
state_dict = torch.load(weights_path) # Load the .pth file

# Extract the actual model weights
Expand Down

0 comments on commit 0e3832b

Please sign in to comment.