MedARC-AI / fMRI-reconstruction-NSD

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

Missing pretrain models in 'train_autoencoder.py' #32

Closed DavisMeee closed 1 year ago

DavisMeee commented 1 year ago

I think there's a missing file for training autoencoder. (No released link for this pretrain model) image

Atom-101 commented 1 year ago

The weights for sd_image_var_autoenc.pth have been added to the HF repo. Check the updated README. Thanks.