Here you can find the CNN model and the script for some of the plots used in the paper: arxiv.org/abs/2009.12872
Due to size limit files are not included here. Readers are encouraged to contact authors at sam.nep (at) gmail.com for the full set of trained CNN models, training spectra & gradients.