You can deploy this model and start earning money today!

batmangiaicuuthegioi/
gae-checkpoints

Name: gae-checkpoints
Availability: InStock

Views

No views yet

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

python

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)