Thank you for your reply!
This is the repository I used:

  
      
      GitHub
  
  
    

xiongzihua/pytorch-YOLO-v1 6

an experiment for yolo-v1, including training and testing. - xiongzihua/pytorch-YOLO-v1


  
  
    
    
  
  


But I tried to call nn.DataParallel() to implement multiple GPUs training follow the instructions below:
https://pytorch.org/tutorials/beginner/blitz/data_parallel_tutorial.html#imports-and-parameters 5
And then I got the error mentioned above, I’m not sure whether it is because the type  of two GPUs  I use is different, but I’m doubting…