Autoencoder models on huggingface have incorrect name

MedARC-AI / fMRI-reconstruction-NSD

fMRI-to-image reconstruction on the NSD dataset.

MIT License

294 stars 39 forks source link

Autoencoder models on huggingface have incorrect name #29

Closed reesekneeland closed 12 months ago

reesekneeland commented 12 months 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 12 months ago

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

DavisMeee commented 11 months 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'.

Do you mind to help me out? Thanks!

reesekneeland commented 11 months ago

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