Question about MAISI training batch size

[Mors20]

Hi,

Thank you for your excellent work on MAISI. I’ve noticed that the model can generate images with flexible dimensions, and I’m curious about the training process that allows for this flexibility. From my understanding, training with images of varying sizes would typically require a batch size of 1, which can be challenging, as batch size is crucial for diffusion model training. It seems the paper doesn't go into detail on this aspect, so I would appreciate any insights you can provide.

Best regards,
Haoshen