dedonnoc/retinal-models-mouse

retina-mouse (binary)

gReLU / Enformer model of mouse retinal chromatin accessibility, 7 cell types. Shared for collaborator review; not a release.

Tasks, in order (positional — on_tasks/off_tasks are indices): AC, BC, Cone, HC, MG, RGC, Rod (= Amacrine, Bipolar, Cones, Horizontal, Muller, RGC, Rods)

Files

file	notes
retina_mouse_binary.ckpt	saved by gReLU 1.0.4
retina_mouse_binary.grelu106.ckpt	same weights, loadable by gReLU >= 1.0.6

gReLU 1.0.4 stores data_params only under hyper_parameters, which >=1.0.6 cannot read. Use the .grelu106 file on modern gReLU — weights are identical.

from grelu.lightning import LightningModel
model = LightningModel.load_from_checkpoint("retina_mouse_binary.grelu106.ckpt")

Supersedes the mouse checkpoints in Genentech/retinal-models. Note the filename retina_mouse_binary.ckpt is a different training run from the one published there — do not treat them as interchangeable.