Codes: https://github.com/qibin2020/caloSim_study/blob/main/VQVAE_L1_optimized_v4_send.ipynb