Code for the pytorch models used in the retinal-rl project (and related non-rl usages) - View it on GitHub