Complex-YOLOv4-Pytorch代码地址:
https://gitcode.net/mirrors/maudzung/Complex-YOLOv4-Pytorch?utm_source=csdn_github_accelerator
1报错:RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1
解决办法:
map_location='cuda:0'
如下图:
版权声明:本文为wss794原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。