Hi @feizc , thanks for your great work. I find that following README's command to train DiS-L/2, even with one image on each GPU, the GPU memory rises to 70G. I think it is abnormal since in the paper it is said DiS is trained with NVIDIA A100. I cannot imagine how to train larger variants beyond L/2.