Could you update to PyTorch 1.5.1 and CUDA10.2.89 + cudnn7.6.5.32?
Also, is the YOLO code working fine and the error is raised by your dataparallel code?
If so, could you post the code to reproduce this error (with input shapes if possible)?