Published July 2, 2026 | Version v1
Model Open

Autoencoder and normalising flow checkpoints.

  • 1. ROR icon Commissariat à l'Énergie Atomique et aux Énergies Alternatives
  • 2. NSF-Simons AI Institute for the Sky
  • 3. Departement of Astronomy and Astrophysics, University of Chicago

Description

The files in this repository correspond to the weights of the pretrained autoencoder and normalising flow used for the work "SHINE: Shear inference environment - Generative forward modelling for radio weak gravitational lensing". The asociate repository for this work is https://github.com/CentofantiEze/Radio_WL_Generative_Forward_Model.

The models are defined in the probabilistic-shear repository: https://github.com/b-remy/probabilistic-shear.

Contents:

- Autoencoder (AE): chekcpoints for the full AE and its distilled versions (half and quarter size). 

- Normalising Flow (NF): checkpoint for the realNVP NF. 

 

Files

model_checkpoints.zip

Files (146.2 MB)

Name Size
md5:84b7bcaeb88a227c39423abcf8a2dc47
146.2 MB Preview Download

Additional details