soinx0629 / vis_dec_neurips

Official Implementation for NeurIPS'23 paper Contrast, Attend and Diffuse to Decode High-Resolution Images from Brain Activities
41 stars 7 forks source link

LDM Model Config File #9

Open loungeterm opened 5 months ago

loungeterm commented 5 months ago

I'm trying to train the LDM myself, but can't find the config file that referenced in the following line of code:

https://github.com/soinx0629/vis_dec_neurips/blob/eb2e419a631862441a521c0bf65961f5b1db49f4/code/dc_ldm/ldm_for_fmri.py#L46-L53

Would it be possible to upload that file to GitHub too? It would also be great if the trained LDM checkpoint was available for comparison

romain-dufly commented 4 months ago

pretrain_root refers to a folder that is only available in the authors previous project, MindVis. You can copy it from https://github.com/zjc062/mind-vis. I tested it and the LDM does get the announced performance when using this folder.