May I ask how many epochs you trained for and how much GPU memory it used?

[ZXX-Coding]

[ControlNet]

We use 2 RTX A6000 48GB GPUs for training. I remember we run out of almost all the vRAM with batch size 16x2, you can run it with less batch size to allow it runs with less vRAM.