apex icon indicating copy to clipboard operation
apex copied to clipboard
verified publisher icon NVIDIA

Parallel wrappers should only be applied to the model(s) AFTER \n

Open jinyaxuan opened this issue 3 years ago • 0 comments

RuntimeError: Incoming model is an instance of torch.nn.parallel.DataParallel. Parallel wrappers should only be applied to the model(s) AFTER the model(s) have been returned from amp.initialize.

How to solve it

jinyaxuan avatar Apr 24 '22 09:04 jinyaxuan