Remove optimizer history from pretrained model checkpoints (#81)

microsoft · Jan 5, 2022 · 0f3ccf8 · 0f3ccf8
1 parent 01a9bd8
commit 0f3ccf8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphormer/pretrain/__init__.py b/graphormer/pretrain/__init__.py
@@ -2,8 +2,8 @@
 import torch.distributed as dist
 
 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",
+    "pcqm4mv1_graphormer_base":"https://szheng.blob.core.windows.net/graphormer/modelzoo/pcqm4mv1/checkpoint_best_pcqm4mv1.pt",
+    "pcqm4mv2_graphormer_base":"https://szheng.blob.core.windows.net/graphormer/modelzoo/pcqm4mv2/checkpoint_best_pcqm4mv2.pt",
     "oc20is2re_graphormer3d_base":"https://szheng.blob.core.windows.net/graphormer/modelzoo/oc20is2re/checkpoint_last_oc20_is2re.pt",
 }