torch2onnx

質問1：トーチモデルには条件付きブランチがあります

ONNXエクスポートの失敗：オペレーターsiluのONNXopsetバージョン12へのエクスポートはサポートされていません。バグを開いて、不足しているオペレーターのONNXエクスポートサポートをリクエストしてください。

 siluの実装を変更するには、リンクを参照してください

class SiLU(Module):
 
    __constants__ = ['inplace']
    inplace: bool
 
    def __init__(self, inplace: bool = False):
        super(SiLU, self).__init__()
        self.inplace = inplace
 
    def forward(self, input: Tensor) -> Tensor:
        # ------------------------------------- #
        # 把F.silu替换掉，修改后如下
        return input * torch.sigmoid(input)
 
        #原来的代码
        return F.silu(input, inplace=self.inplace)

2 trt

ERROR: builtin_op_importers.cpp:1602 In function importIf:
[8] Assertion failed: cond.is_weights() && cond.weights().count() == 1 && "If condition must be a initializer!"

来た

There is no registered "infer" function for node "If_156" with op = "If". Please implement this function in the extensions.

リンクを参照

リンク

importIf、ifノードでエラーが発生しましたが、forward関数に判定を書き込みませんでした。onnxの出力を調べたところ、squeeze（）関数に問題があることがわかりました。torch1.7バージョンでは判定操作が追加されました。 onnxで、スクイーズの次元が1であるかどうかを判断する必要があります。