【报错】运行Complex-YOLO报错处理

Complex-YOLOv4-Pytorch代码地址:
https://gitcode.net/mirrors/maudzung/Complex-YOLOv4-Pytorch?utm_source=csdn_github_accelerator

watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBASmFja2lsaW5hX1N0b25l,size_13,color_FFFFFF,t_70,g_se,x_16

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版权协议,转载请附上原文出处链接和本声明。