Giters

lsj2408 / Transformer-M

[ICLR 2023] One Transformer Can Understand Both 2D & 3D Molecular Data (official implementation)

Home Page:https://arxiv.org/abs/2210.01765

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to load checkpoint of transformer M

santanusantra opened this issue · comments

commented

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

commented

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)