TNN_Demo icon indicating copy to clipboard operation
TNN_Demo copied to clipboard

您好,模型转为tnn 精度下降问题

Open duoduo1018 opened this issue 3 years ago • 2 comments

您好, 我用官方给的yolov5s.pt转为tnn,测试结果正常。 但把自己训练的pt转为tnn后,精度下降很多(跟pytorch版本对比),这是哪里的原因呢?

duoduo1018 avatar Jul 23 '21 08:07 duoduo1018

这个可能需要给官方提issues

cmdbug avatar Jul 23 '21 11:07 cmdbug

您好, 我用官方给的yolov5s.pt转为tnn,测试结果正常。 但把自己训练的pt转为tnn后,精度下降很多(跟pytorch版本对比),这是哪里的原因呢?

https://github.com/ppogg/shufflev2-yolov5

试一下这个把兄弟,我也想知道这个模型转TNN或者MNN会不会崩

ppogg avatar Aug 31 '21 03:08 ppogg