Open LitLeo opened 2 years ago
解决了PReLU层 slope参数的维度转换错误。 对应的issue:https://github.com/htshinichi/caffe-onnx/issues/21 pytorch 转 onnx 之前也有这个错误,fixed详见 https://github.com/pytorch/pytorch/pull/21330
原先只支持PReLU层的输入维度为固定4维,即将slope的维度设置为[dim, 1, 1] 修改后,支持输入为变化维度。
修改思路为在修改param shape时,将input_shape作为参数输入,并根据input_shape决定slope的维度。
解决了PReLU层 slope参数的维度转换错误。 对应的issue:https://github.com/htshinichi/caffe-onnx/issues/21 pytorch 转 onnx 之前也有这个错误,fixed详见 https://github.com/pytorch/pytorch/pull/21330
原先只支持PReLU层的输入维度为固定4维,即将slope的维度设置为[dim, 1, 1] 修改后,支持输入为变化维度。
修改思路为在修改param shape时,将input_shape作为参数输入,并根据input_shape决定slope的维度。