tosonw

好的，麻烦您了

@LVROBOT 不能

@CaoWGG 我使用官方.cfg和.weigth时是没问题的

@CaoWGG 我使用自己的数据集训练了一个2分类，就出了问题

@CaoWGG 我保存模型的代码：torch.save(model.state_dict(), 'logs/Epoch%d-Total_Loss%.4f-Val_Loss%.4f.pth'%((epoch+1),total_loss/(epoch_size+1),val_loss/(epoch_size_val+1))) model就是模型结构，这个我对比过，除了命名不同，其他通道和结构等没有问题

@Happykerry 微调？？是修改最后输出维度吗？

@CaoWGG 我开始用的的官方的yolov4_weights.pth，转换成功了

> @tosonw 权重微调，我用torch存出来的模型 用darknet啥也检不到 @HappyKerry 我测过保存出的pth权重，检测完全正常

> @tosonw > 额，可能它就把yolov4.weights换了个名字成yolov4-weights.pth。 > 从保存的格式上来说，.pth是不会被darknet框架正常读取的 @CaoWGG 好的，麻烦您了，我再去研究下

> > @tosonw > > 额，可能它就把yolov4.weights换了个名字成yolov4-weights.pth。 > > 从保存的格式上来说，.pth是不会被darknet框架正常读取的 > > @CaoWGG 好的，麻烦您了，我再去研究下 @CaoWGG 可是我自己训练出来的centernet的pth文件使用您TensorRT-centernet是可以转换的