search

lsh0520 / 3D-MoLM

39 stars 5 forks source link

Stage1 and stage2 checkpoints #7

Closed icycookies closed 6 months ago

icycookies commented 6 months ago

Thanks for your efforts in releasing this inspiring work! However, it seems that the provided generalist model is the checkpoint after stage3 pre-training. I wonder if the authors could provide the pre-trained checkpoints after stage1 and stage2 for evaluating molecule-text retrieval and molecule captioning.

lsh0520 commented 6 months ago

Thanks for your interest in our work. Will release them in a week.

lsh0520 commented 6 months ago

Hi, we haved uploaded stage1 & stage2 ckpts to huggingface. Please check the Checkpoints section in updated README file to reproduce all the results (Table 2&3&4) in the paper.

icycookies commented 6 months ago

Thanks for your quick reply!