yolov5中train.py报错RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory
时间: 2023-12-31 19:21:46 浏览: 173
yolo-使用onnxruntime部署yolov5目标检测算法.zip
该错误通常是由于PyTorch无法读取权重文件导致的。可能的原因是权重文件已损坏或不完整。您可以尝试重新下载权重文件并确保它完整无损。另外,您还可以尝试使用map_location参数将权重文件加载到CPU上,以避免CUDA内存泄漏。具体来说,您可以在torch.load()函数中使用以下代码:map_location='cpu'。
阅读全文