update weights with 60th epoch ckpt (#6)

ckpt 512x512 training on 26th epoch w/o gradient accumulation (#5)

update the diffusion model weights trained up to 20th epoch (#2)

12th epoch checkpoint on 512x512 datasets (#1)