RECONS_NAME = "recons.jpg" LATENTS_NAME = "latents.pt" CATTN_NAME = "cattn.pt" CATTN_VIZ_NAME = "cattn.jpg"