ICS checkpoints

Checkpoints for ICS (Input-Conditioned Search; code: https://github.com/icsearch/ICS) and the baselines.

Layout

<model>/<task>/seed<k>/ holds config.json and model.safetensors.

Loading

from huggingface_hub import snapshot_download

root = snapshot_download(repo_id="hcshi/ICS", allow_patterns="trm/maze/seed0/*")

With the code installed, python -m ics download --model trm --task maze --seed 0 writes the same folder to ckpt/trm/maze/seed0.

License

MIT.