Skip to content

Commit 33c6e6b

Browse files
committed
Update: a new 50g ckpts, HF Hub release
1 parent f159e55 commit 33c6e6b

2 files changed

Lines changed: 6 additions & 0 deletions

File tree

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
[![pytorch](https://img.shields.io/badge/PyTorch_2.1+-ee4c2c?logo=pytorch&logoColor=white)](https://pytorch.org/get-started/locally/)
66
[![pyg](https://img.shields.io/badge/PyG_2.4+-3C2179?logo=pyg&logoColor=#3C2179)](https://pytorch-geometric.readthedocs.io/en/latest/install/installation.html)
77
[![arxiv](http://img.shields.io/badge/arxiv-2310.04562-yellow.svg)](https://arxiv.org/abs/2310.04562)
8+
[![HuggingFace Hub](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Models-black)](https://huggingface.co/collections/mgalkin/ultra-65699bb28369400a5827669d)
89
![license](https://img.shields.io/badge/License-MIT-green.svg?labelColor=gray)
910

1011
</div>
@@ -47,6 +48,11 @@ Table of contents:
4748
* [Datasets](#datasets)
4849
* [Adding custom datasets](#adding-your-own-graph)
4950

51+
## Updates
52+
53+
* **Dec 4th, 2023**: Added a new ULTRA checkpoint `ultra_50g` pre-trained on 50 graphs. Averaged over 16 larger transductive graphs, it delivers 0.389 MRR / 0.549 Hits@10 compared to 0.329 MRR / 0.479 Hits@10 of the `ultra_3g` checkpoint. The inductive performance is still as good! Use this checkpoint for inference on larger graphs.
54+
* **Dec 4th, 2023**: Pre-trained ULTRA models (3g, 4g, 50g) are now also available on the [HuggingFace Hub](https://huggingface.co/collections/mgalkin/ultra-65699bb28369400a5827669d)!
55+
5056
## Installation ##
5157

5258
You may install the dependencies via either conda or pip.

ckpts/ultra_50g.pth

2.03 MB
Binary file not shown.

0 commit comments

Comments
 (0)