Note this warning may be related to loading older models. You can update your model to current structure with:
import torch
ckpt = torch.load(“model.pt”) # applies to both official and custom models
torch.save(ckpt, “updated-model.pt”)
yolo错误记录(未解决)
最新推荐文章于 2024-10-06 15:50:38 发布
本文介绍了如何使用PyTorch中的torch.load()函数加载旧模型,并提供了解决方案以便将其结构更新到最新版本,适用于官方和自定义模型。torch.save()函数用于保存更新后的模型。
摘要由CSDN通过智能技术生成