From 49de0f57359035d9ae7a69bd989eb712fb047ca2 Mon Sep 17 00:00:00 2001
From: YouJiacheng <83971976+YouJiacheng@users.noreply.github.com>
Date: Wed, 5 Jan 2022 00:14:44 +0800
Subject: [PATCH] Add OC20 IS2RE pretrain checkpoint

---
 graphormer/pretrain/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/graphormer/pretrain/__init__.py b/graphormer/pretrain/__init__.py
index 3213732..ff039dc 100644
--- a/graphormer/pretrain/__init__.py
+++ b/graphormer/pretrain/__init__.py
@@ -4,6 +4,7 @@
 PRETRAINED_MODEL_URLS = {
     "pcqm4mv1_graphormer_base":"https://szheng.blob.core.windows.net/graphormer/modelzoo/pcqm4mv1/checkpoint_best_pcqm4mv1_full.pt",
     "pcqm4mv2_graphormer_base":"https://szheng.blob.core.windows.net/graphormer/modelzoo/pcqm4mv2/checkpoint_best_pcqm4mv2_full.pt",
+    "oc20is2re_graphormer3d_base":"https://szheng.blob.core.windows.net/graphormer/modelzoo/oc20is2re/checkpoint_last_oc20_is2re.pt",
 }
 
 def load_pretrained_model(pretrained_model_name):