18910140161

我把yolov4的网络结构改了,预训练权重用不了了,有什么方法解决吗?

顺晟科技

2021-07-03 21:01:29

172

先找到权重字典中那些和原来网络不同的那个键值对,del删掉,然后在加载权重的时候设置 strict=False

重新训练。。。

相关文章
我们已经准备好了,你呢?
2024我们与您携手共赢,为您的企业形象保驾护航