MedARC-AI / MindEyeV2

MIT License
120 stars 20 forks source link

Question about Contrastive Learning Collapse #38

Open EdgarAllanPoo opened 1 month ago

EdgarAllanPoo commented 1 month ago

Thank you very much for your work. May I ask have you ever encounter a condition during experiments where your model collapses (only able to reconstruct certain classes only)? I have tried to replicate your work as a basis on other datasets . However, my experiments only able to reconstruct 2 types of images regardless of the pair in the test set. Looking forward to your reply!