search

lsj2408 / Transformer-M

[ICLR 2023] One Transformer Can Understand Both 2D & 3D Molecular Data (official implementation)
https://arxiv.org/abs/2210.01765
MIT License
201 stars 24 forks source link

How to load checkpoint of transformer M #6

Closed santanusantra closed 1 year ago

santanusantra commented 1 year ago

Great piece of work. Could you please mention, how to load your checkpoint?

lsj2408 commented 1 year ago

Thanks for your interest in our work! To load the checkpoints, you need to first download the corresponding checkpoints and then set the path to the checkpoints as "ckpt_path" (for fine-tuning) or "save_dir" (for evaluation on PCQM4Mv2)