gridfm/genco-opf-datakit

GENCO optimal-power-flow checkpoints on datakit grids (paper): Base and Small, IEEE 14, 30, 57, and 118, and GOC 500 and 2000, seeds 0 and 1.

Evaluate on the paper branch by following GENCO §5.4.3. That guide uses graphkit genco-paper-repro and is how the paper numbers were produced.

Or evaluate the same checkpoints on graphkit main. main is under development, so the same numbers are not guaranteed. They matched the paper-branch result at commit f104900 (1 October 2026).

git clone https://github.com/gridfm/gridfm-graphkit.git
cd gridfm-graphkit
pip install -e .
pip install "huggingface_hub[cli]"
TORCH_CUDA_VERSION=$(python -c "import torch; print(torch.__version__ + ('+cpu' if torch.version.cuda is None else ''))")
pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH_CUDA_VERSION}.html
export MLFLOW_ALLOW_FILE_STORE=true

mkdir -p data/case118_ieee/raw
hf download gridfm/opf_small_case118_ieee --repo-type dataset --local-dir data/case118_ieee/raw
hf download gridfm/genco-opf-datakit --include "case118_ieee/base/seed0/**" --local-dir genco-opf-datakit

gridfm_graphkit evaluate \
  --config_conversion auto_inline \
  --config genco-opf-datakit/case118_ieee/base/seed0/case118_ieee_base_seed0.yaml \
  --data_path data \
  --model_path genco-opf-datakit/case118_ieee/base/seed0/last.pt \
  --normalizer_stats genco-opf-datakit/case118_ieee/base/seed0/normalizer_stats.pt

Pass last.pt, except GOC 2000 seed 0, where the model path is best_model_state_dict.pt.