Trained checkpoints https://drive.google.com/drive/folders/1DNLuxHkwhHLVprL9R_b8mmhDaIpAgOJe?usp=sharing