如何转换成.pt文件？

[lkx]

pytorch版本中的导出模型只有onnx一种，在我加了Adaptavgpool2d()操作之后，由于onnx中没有这一操作子，导致代码报错，所以请问一下如何将其导出为.pt文件。
或者在公平对比几个模型的fps时，我应该如何对比？
非常感谢解答！

[lyuwenyu]

在我加了Adaptavgpool2d()操作之后，由于onnx中没有这一操作子

1. 修改后的模型？ 推理时候固定输入的话 Adaptavgpool2d可以用Avgpool2d平替吧
2. 模型初始化之后 torch.save 模型.pt，详细可以查看torch的文档