Graphormer icon indicating copy to clipboard operation
Graphormer copied to clipboard

Can't load the pretrain model

Open wen19shui opened this issue 1 year ago • 7 comments

Hi, I failed in load the pretrain model ,here about the error : urllib.error.URLError: <urlopen error [Errno -2] Name or service not known> and then I want to get the url pretrain model's url the links I got from /pretrain/_init_.py seems to be shut down. The reason for closing is no longer to share. Could you please share the pre training model again? Thank you very much.

wen19shui avatar Jul 27 '22 09:07 wen19shui

I also have the same problem.

yuanbatou avatar Jul 29 '22 09:07 yuanbatou

same problem as you

Frankewu0 avatar Jul 30 '22 13:07 Frankewu0

Hi, sorry for the late response. We're working on moving the pretrained models to a new site, and they will be available soon, please stay tuned.

mavisguan avatar Aug 04 '22 01:08 mavisguan

To fix the problem, you can change pcqm4mv1 pretrained model's url to https://ml2md.blob.core.windows.net/graphormer-ckpts/checkpoint_best_pcqm4mv1.pt in file graphormer/pretrain/__init__.py. We'll update the code soon.

mavisguan avatar Aug 04 '22 07:08 mavisguan

Thank you for your fix. I will try it soon. By the way, if I want to train a 3D model with the help of Graformer_3D, do I need to use a specific pretrained model, thanks a lot.

wen19shui avatar Aug 04 '22 07:08 wen19shui

For now you can use our oc20 pretrained model (which will be moved to the new blob container soon), but we can't guarantee best performance because it was pretrained on the OC20 is2re task using a relatively small dataset. In near future we plan to release graphormer-3d model pretrained on a large-scale dataset.

mavisguan avatar Aug 05 '22 01:08 mavisguan

@mavisguan It's very exciting that in near future you plan to release graphormer-3d model pretrained on a large-scale dataset. Is this coming anytime soon? I'm not finding any graphormer-3d pretrained models here

https://github.com/microsoft/Graphormer/blob/77f436db46fb9013121289db670d1a763f264153/graphormer/pretrain/init.py#L4

jozhang97 avatar Apr 27 '23 14:04 jozhang97