Nioolek / PPYOLOE_pytorch

An unofficial implementation of Pytorch version PP-YOLOE,based on Megvii YOLOX training code.
Apache License 2.0
180 stars 20 forks source link

您使用的pytorch版本是多少? #9

Open lixiangMindSpore opened 2 years ago

Nioolek commented 2 years ago

pytorch 1.6.0、1.7.0、1.7.1都进行过测试可以适配。 但目前半精度训练会有一定概率导致精度为0

lixiangMindSpore commented 2 years ago

pytorch 1.6.0、1.7.0、1.7.1都进行过测试可以适配。 但目前半精度训练会有一定概率导致精度为0

如果使用1.10.0可以训练和模型转换(paddle->pytorch)吗? image

Nioolek commented 2 years ago

pytorch 1.6.0、1.7.0、1.7.1都进行过测试可以适配。 但目前半精度训练会有一定概率导致精度为0

如果使用1.10.0可以训练和模型转换(paddle->pytorch)吗? image

应该没有大的问题。如果在使用pytorch1.10.0出现一些问题,可以继续提交issues

lixiangMindSpore commented 2 years ago

pytorch 1.6.0、1.7.0、1.7.1都进行过测试可以适配。 但目前半精度训练会有一定概率导致精度为0

如果使用1.10.0可以训练和模型转换(paddle->pytorch)吗? image

应该没有大的问题。如果在使用pytorch1.10.0出现一些问题,可以继续提交issues

感谢专家,帮忙看一下这个 issue:https://github.com/Nioolek/PPYOLOE_pytorch/issues/10

lixiangMindSpore commented 2 years ago

pytorch 1.6.0、1.7.0、1.7.1都进行过测试可以适配。 但目前半精度训练会有一定概率导致精度为0

如果使用1.10.0可以训练和模型转换(paddle->pytorch)吗? image

应该没有大的问题。如果在使用pytorch1.10.0出现一些问题,可以继续提交issues

专家,麻烦再看一下这个,多谢 https://github.com/Nioolek/PPYOLOE_pytorch/issues/11