请问如何进行多卡训练呢？ #15
Description
您好，感谢您的elegant工作，
想询问下我在调整分辨率到512后训练batch_size=1（A100）需要52gb的显存，2.5小时每epoch
在训练batch_zize=2时显存爆掉
然后希望使用多卡训练，该如何使用呢？
{{ message }}
您好，感谢您的elegant工作，
想询问下我在调整分辨率到512后训练batch_size=1（A100）需要52gb的显存，2.5小时每epoch
在训练batch_zize=2时显存爆掉
然后希望使用多卡训练，该如何使用呢？