Hi, I feel sorry to bother you again, but after a long time simulation, the results seem to be incorrect. Do you have any idea of the reason? I think there should be no other bugs as the program is simple and the result is correct when I use single GPU. Thanks again.
With single GPU, I comment the following line
model=nn.DataParallel(model, device_ids=....)