Inquiry of pretrained checkpoint

Question

Inquiry of pretrained checkpoint

OliverRensu opened this issue 2 years ago · 3 comments

Thanks for your fantastic work! May I ask when you will release the pretrained checkpoint?

Answer 1 · 2023-06-06T03:25:29.000Z

Thanks for waiting. The pretrained model is now on hugglingface Pretrained model download, you can also download it with:

from huggingface_hub import snapshot_download
models_path = snapshot_download("shgao/MDT-XL2")
ckpt_model_path = os.path.join(models_path, "mdt_xl2_v1_ckpt.pt")

A hugglingface demo is on DEMO.

Answer 2 · 2024-03-14T03:03:33.000Z

@gasvn Can you please also provide the pretrained checkpoints of MDT-S2 and MDT-B2?

Answer 3 · 2024-03-14T14:14:20.000Z

Sorry, I don't have the ckpt for s2/b2 now becasue I no longer have the access to the server for training them. The training cost for s2/b2 is relatively small.