MedARC-AI / MindEyeV2

MIT License
113 stars 18 forks source link

Question about Contrastive Learning Collapse #38

Open EdgarAllanPoo opened 1 day ago

EdgarAllanPoo commented 1 day 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!