使用两颗GPU复现NAFNet-width32疑问

megvii-research / NAFNet

The state-of-the-art image restoration model without nonlinear activation functions.

Other

2.18k stars 267 forks source link

使用两颗GPU复现NAFNet-width32疑问 #107

Open Barry-CHIU opened 1 year ago

Barry-CHIU commented 1 year ago

作者您好

我在先前的提问中看到作者是用8颗GPU训练，而当使用2颗GPU训练时，是否要将training iteraation增加到原始设定的4倍呢? 我尝试将training iteration增加4倍，但达不到paper中GoPro dataset的数字 (差0.3dB) 不过把iteration数增加到8倍后，就超过paper当中的数字 (多约0.2 dB) 不太确定怎么样的设定才是对的

kk047ddd commented 1 year ago

您好，请问您的batch_size_per_gpu跟原来一样设为4吗？