文章/答案/技术大牛

发布

社区首页 >问答首页 >喷炬预训练模型重量d型的变化

问喷炬预训练模型重量d型的变化
EN

Stack Overflow用户

提问于 2022-08-23 06:36:18

回答 1查看 231关注 0票数 0

我使用的是YOLOV7模型。预训练权值共享在float16 dtype中被优化和共享。

如何在PyTorch中转换模型参数的d型。我想将权重类型转换为float32类型。

weights = torch.load('yolov7-mask.pt')
model = weights['model']

回答已采纳

发布于 2022-08-25 08:59:29

将权重加载到模型中，只需调用.float()即可。

示例：

cp = torch.load('yolov7-mask.pt')
model.load_state_dict(cp['weight'])
model = model.float()

如果模型的类是nn.Module，它就能工作。(检查火炬版本1.8)

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/73454134

复制

相似问题

问喷炬预训练模型重量d型的变化EN