RainbowPiBubbles

I keep running out of memory when running GaussianDiffusion1D. Is there a way to change the batch size?

1

model = Unet1D( dim = 64, dim_mults = (1, 2, 4, 8), channels = 1, ) diffusion = GaussianDiffusion1D( model, seq_length = 128, timesteps = 250, objective = 'pred_v', )...