1.3.2 版本， pytorch的 ReLU6 op 不支持

这种一般都是后缀版本号造成不支持。nn.ReLU 和 ReLU6有啥区别么？可以换么？

kkkaaa 发表于 2020-5-13 15:03
的确可以替换，可以绕过这个问题

现实一点的问题是，使用 relu6 训练好了一个模型, 模型有5个输出，通过 ...

我没研究过pytorch，不太懂他的relu6和relu有啥区别。。
换relu后最好还是要重新训练下。
你说的结果精度误差，是可以通过混合量化来解决的，这个不是问题。

确实是个问题。。
我这里只能给NPU部门报这个op，他们支持起来最快最快也得下个版本才能搞完，时间上会更久。
建议你同一个模型可以用TF试试，毕竟pytorch刚开始支持。。

NPU Team回复：

该OP已经在下一个版本支持，请等待下一个版本发布