lyuwenyu
commented
6 months ago 在我加了Adaptavgpool2d()操作之后，由于onnx中没有这一操作子
- 修改后的模型？ 推理时候固定输入的话
Adaptavgpool2d可以用Avgpool2d平替吧 - 模型初始化之后
torch.save模型.pt，详细可以查看torch的文档
Open lkxxxxx opened 7 months ago
lyuwenyu
commented
6 months ago 在我加了Adaptavgpool2d()操作之后，由于onnx中没有这一操作子
Adaptavgpool2d可以用Avgpool2d平替吧torch.save 模型.pt，详细可以查看torch的文档
pytorch版本中的导出模型只有onnx一种，在我加了Adaptavgpool2d()操作之后，由于onnx中没有这一操作子，导致代码报错，所以请问一下如何将其导出为.pt文件。 或者在公平对比几个模型的fps时，我应该如何对比？ 非常感谢解答！