search

MedARC-AI / fMRI-reconstruction-NSD

fMRI-to-image reconstruction on the NSD dataset.
MIT License
307 stars 44 forks source link

Autoencoder models on huggingface have incorrect name #29

Closed reesekneeland closed 1 year ago

reesekneeland commented 1 year ago

Hello,

The pretrained autoencoder for subject1 is called "epoch120.pth", which is what the code expects, however the pretrained models for subjects 2, 5, and 7 are called "last.pth", causing an error when the model is loaded in the Reconstructions script.

PaulScotti commented 1 year ago

Thank you! I have changed the names for subj 2 5 7 to be epoch120 instead of last.pth on huggingface

DavisMeee commented 1 year ago

Hey Reese,

Sorry to bother you. But I want to know have you tried to train that autoencoder through 'train_autoencoder.py'? It seems that it has missed one file named 'sd_image_var_autoenc.pth'. image

Do you mind to help me out? Thanks!

reesekneeland commented 1 year ago

Hello, I have only been using the pretrained models so I have not tested the training script at all.