size_from_dim: Assertion `dim ＞= 0 && (size_t)dim ＜ sizes_.size()` failed.

遇到的问题

/home/optimizer-master/third_party/onnx_common/tensor.h:117: size_from_dim: Assertion dim >= 0 && (size_t)dim < sizes_.size() failed.

问题描述

在导onnx的时候，遇到了上述问题

解决办法

onnx导出的时候存在不能导出的op操作，就会报这个错
解决办法：
找到不能导出op的操作，增加一个判断如果进行导出onnx，则不使用该op,例子如下

def forward(self, x):
    if torch.onnx.is_in_onnx_export():#如果进行到onnx
        return x * self.act(x) / 6
    else: # 正常训练
        return x * self.act(x + 3) / 6