when i use the code “net = torch.nn.SyncBatchNorm.convert_sync_batchnorm(net)” to replace BN with SyncBatchNorm, the code would be deadlock like this:
image880×293 30.8 KB
it seems to be a problem with dataloader. And the relevant code is as follows
image1114×559 69.8 KB
image1066×480 45.7 KB
Is there any kind person to help me？Thanks.