EleGANt icon indicating copy to clipboard operation
EleGANt copied to clipboard

Published 20 hours ago • verified publisher icon Chenyu-Yang-2000

请问如何进行多卡训练呢？

Open 1605707467qq opened this issue 1 year ago • 0 comments

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

1605707467qq avatar Nov 06 '23 08:11 1605707467qq