batmangiaicuuthegioi/refined-gae-checkpoints

Refined-GAE Model for ogbl-collab

This model was trained using the Refined-GAE framework.

Model Details

• Model: GCN
• Dataset: ogbl-collab
• Run Name: collab-raw-feat-and-embed
• Hidden Channels: 1024
• Epochs: 20
• Propagation Steps: 4

Usage

import torch
from huggingface_hub import hf_hub_download

checkpoint_path = hf_hub_download(
    repo_id="batmangiaicuuthegioi/refined-gae-checkpoints",
    filename="checkpoints/ogbl-collab_GCN_collab-raw-feat-and-embed.pt"
)

state_dict = torch.load(checkpoint_path)