KaryoFlow / reproducibility /weights.md
procaffenator's picture
License original contributions and trained checkpoints under Apache 2.0
6e36ea7 verified
|
Raw History Blame Contribute Delete
6.06 kB

Model weights

Download checkpoints from KaryoFlow on Hugging Face. The files are provided individually; no archive extraction is needed. Place them under weights/ in the code repository.

Checkpoint downloads

The seed labels below refer to training seeds. Dataset 2 comparison detectors use one trained checkpoint per model, evaluated with three inference seeds.

Model Dataset 1 Dataset 2
KaryoFlow Seed 42 · Seed 123 · Seed 789 Seed 42 · Seed 123 · Seed 789
KaryoFlow + LQCR Seed 42 · Seed 123 · Seed 789 Seed 42 · Seed 123 · Seed 789
KaryoFlow H3 Seed 42 · Seed 123 · Seed 789 Seed 42 · Seed 123 · Seed 789
Cascade R-CNN Seed 42 · Seed 123 · Seed 789 Checkpoint
DiffusionDet Seed 42 · Seed 123 · Seed 789 Checkpoint
RTMDet-L Seed 42 · Seed 123 · Seed 789 Checkpoint
YOLOX-S Seed 42 · Seed 123 · Seed 789 Checkpoint

Euler and Heun reuse the corresponding KaryoFlow checkpoints. H3 files contain student parameters only. The configuration and result index maps all 54 recorded evaluations to 34 distinct checkpoint files.

Download from the command line

Run from the code repository root:

hf download procaffenator/KaryoFlow \
  weights/d2_karyoflow_lqcr_seed42.pth --local-dir .

Download all checkpoints:

hf download procaffenator/KaryoFlow --include "weights/*.pth" --local-dir .
python tools/check_release.py --weights

For one-model evaluation, only its selected checkpoint is needed. Reproducing a three-training-run mean requires the three corresponding checkpoints.

Data and license scope

Dataset 1 images are not distributed; its checkpoints do not provide access to the cohort. Dataset 2 must be obtained from its original source as described in data preparation.

The contributors’ rights in these checkpoints are licensed under Apache-2.0. See Checkpoint licensing for the covered files and retained upstream rights, and License scope for code and data terms.

The checkpoint file sizes include buffers and serialization overhead. They differ from the FP32 parameter-payload sizes reported in the paper.