pytorch-yolov3 train 报错

本文详细介绍了在PyTorch-YOLOv3项目中,如何将uint8类型的张量转换为bool类型,以适应新的PyTorch版本要求。通过将obj_mask和noobj_mask字段从int8类型更改为bool类型,可以避免警告信息并确保模型的正确运行。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

"D:\postgraduate\python\my experiment\yolov9\.venv\Scripts\python.exe" "D:\postgraduate\python\my experiment\ssd-pytorch-master(1)\ssd-pytorch-master\train.py" initialize network with normal type Load weights D:\postgraduate\python\my experiment\ssd-pytorch-master(1)\ssd-pytorch-master\model_data\ssd_weights.pth. Successful Load Key: [&#39;vgg.0.weight&#39;, &#39;vgg.0.bias&#39;, &#39;vgg.2.weight&#39;, &#39;vgg.2.bias&#39;, &#39;vgg.5.weight&#39;, &#39;vgg.5.bias&#39;, &#39;vgg.7.weight&#39;, &#39;vgg.7.bias&#39;, &#39;vgg.10.weight&#39;, &#39;vgg.10.bias&#39;, &#39;vgg.12.weight&#39;, &#39;vgg.12.bias&#39;, &#39;vgg.14.weight&#39;, &#39;vgg.14.bias&#39;, &#39;vgg.17.weight&#39;, &#39;vgg.17.bias&#39;, &#39;vgg.19.weight&#39;, &#39;vgg.19.bias&#39;, &#39;vgg.21.weight&#39;, &#39;vgg.21.bias&#39;, &#39;vgg.24.weight&#39;, &#39;vgg.24.bias&#39;, &#39;vgg.26.weight&#39;, &#39;vgg.26.bias&#39;, &#39;vgg.28.weight&#39;, &#39;vgg.28.bias&#39;, &#39;vgg.31.weight&#39;, &#39;vgg.31.bias&#39;, &#39;vgg.33.weight&#39;, &#39;vgg.33.bias&#39;, &#39;extras.0.weight&#39;, &#39;extras. …… Successful Load Key Num: 59 Fail To Load Key: [&#39;conf.0.weight&#39;, &#39;conf.0.bias&#39;, &#39;conf.1.weight&#39;, &#39;conf.1.bias&#39;, &#39;conf.2.weight&#39;, &#39;conf.2.bias&#39;, &#39;conf.3.weight&#39;, &#39;conf.3.bias&#39;, &#39;conf.4.weight&#39;, &#39;conf.4.bias&#39;, &#39;conf.5.weight&#39;, &#39;conf.5.bias&#39;] …… Fail To Load Key num: 12 温馨提示,head部分没有载入是正常现象,Backbone部分没有载入是错误的。 Epoch 1/150: 0%| | 0/151 [00:00<?, ?it/s<class &#39;dict&#39;>]Configurations: ---------------------------------------------------------------------- | keys | values| ---------------------------------------------------------------------- | classes_path | D:\postgraduate\python\my experiment\ssd-pytorch-master(1)\ssd-pytorch-master\model_data\class.txt| | model_path | D:\postgraduate\python\my experiment\ssd-pytorch-master(1)\ssd-pytorch-master\model_data\ssd_weights.pth| | input_shape | [300, 300]| | Init_Epoch | 0| | Freeze_Epoch | 50| | UnFreeze_Epoch | 150| |
最新发布
03-28
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值