I used training_batch_size=3 for 32x32, and this led to the best convergence on my dataset(size=700 images, augmentation can’t be done). Now, I wanted to scale up the resolution to 512x512 . Using the same parameters and I am not getting convergence at, say, after 200 epochs. What should be ideal params values for DDPM training on small high res. batch ? The args of the script are the same as of the official example Training Unconditional.