You can deploy this model and start earning money today!

TorchLight/
DRAFTS

Views

No views yet

DRAFTS Model Repository

Pretrained model weights for DRAFTS, a deep-learning radio fast transient search pipeline.

Current search runtime

These are the two checkpoints used by the current DRAFTS observation-search and injection-search entry points:

Task	Architecture	File	Size	SHA-256
Candidate localization	CenterNet + ConvNeXt-Tiny, v10	`object_best_model_centernet_conv_tiny_ema_v10.pth`	114,449,795 bytes	`bcad4e710f5f1ccd3c8609d35a8d3fbfc36abd1d85bfefd035e945a573fb0629`
Candidate verification	ConvNeXt-Small binary classifier	`binary_best_model_conv_small_ema.pth`	197,949,537 bytes	`2055745aab76ddc16074516aa7b9aafdfaedf16df37ce8924389573eab27ffd8`

Download both files with the Hugging Face CLI:

bash

mkdir -p runcode/models
hf download TorchLight/DRAFTS \
  object_best_model_centernet_conv_tiny_ema_v10.pth \
  binary_best_model_conv_small_ema.pth \
  --local-dir runcode/models

Or use direct HTTP downloads:

bash

curl -L \
  -o runcode/models/object_best_model_centernet_conv_tiny_ema_v10.pth \
  https://huggingface.co/TorchLight/DRAFTS/resolve/main/object_best_model_centernet_conv_tiny_ema_v10.pth
curl -L \
  -o runcode/models/binary_best_model_conv_small_ema.pth \
  https://huggingface.co/TorchLight/DRAFTS/resolve/main/binary_best_model_conv_small_ema.pth

The injection runtime uses the same filenames under injection_experiment/search_runtime/models/. Compare the downloaded files with the SHA-256 values above before running a long search.

Model architecture definitions, preprocessing, command-line examples, and deployment notes are maintained in the DRAFTS repository:

Legacy checkpoints

Earlier ResNet detector and classifier checkpoints are grouped under legacy/. They remain available for reproducing older DRAFTS experiments, but they are not used by the current runtime.

Task	File
CenterNet ResNet-18 detector	`legacy/cent_resnet18.pth`
CenterNet ResNet-50 detector	`legacy/cent_resnet50.pth`
ResNet-18 binary classifier	`legacy/class_resnet18.pth`
ResNet-50 binary classifier	`legacy/class_resnet50.pth`

License and citation

The accompanying DRAFTS code is released under the MIT License. If DRAFTS is useful in your work, cite DRAFTS: A Deep Learning-Based Radio Fast Transient Search Pipeline and record the exact checkpoint and search configuration used.