batmangiaicuuthegioi/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-depth-sweep-5
• Hidden Channels: 256
• Epochs: 20
• Propagation Steps: 2

Usage

import torch
from huggingface_hub import hf_hub_download

checkpoint_path = hf_hub_download(
    repo_id="batmangiaicuuthegioi/gae-checkpoints",
    filename="checkpoints/ogbl-collab_GCN_collab-depth-sweep-5.pt"
)

state_dict = torch.load(checkpoint_path)