models icon indicating copy to clipboard operation
models copied to clipboard

Can it be used in DDP?

Open chengjianhong opened this issue 3 years ago • 0 comments

Hi, I use the GardNorm in my segmentation and classification task. I want to use the DistributedDataParallel to train it. But it occurs the error: "RuntimeError: derivative for batch_norm_backward_elemt is not implemented". Can you give me some advice?

chengjianhong avatar Apr 24 '21 03:04 chengjianhong