search

MedARC-AI / fMRI-reconstruction-NSD

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

Missing pretrain models in 'train_autoencoder.py' #32

Closed DavisMeee closed 11 months ago

DavisMeee commented 11 months ago

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

Atom-101 commented 11 months ago

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