This repository contains the trained checkpoints for the D4LensPINN
gravitational lens classification model and its ResNet-18 baseline.
These weights are provided for reproducibility and research use in
mechanistic interpretability experiments on hybrid physics-ML
architectures.
1import torch
2
3ckpt = torch.load("d4phase2best.pth", map_location="cpu")
The exact model class definitions must match the training code used to
create the checkpoints.
If you use these weights, please cite the associated paper.
Apache-2.0 License.
