Skip to content

ConvTranspose层的channel问题 #43

@lqian

Description

@lqian

在使用onnx2caffe项目转换ConvTranspose2d层到caffe的Deconvolution层时,_operators.py的代码
num_output=W.shape[1],
但我在转的onnx模型中,ConvTranspose2d层的weight的shape是[256, 1, 2, 2],模型源码的定义是
self.deconv = nn.ConvTranspose2d(c1, c1, kernel_size=k, stride=s, padding=0, output_padding=0, groups=c1, bias=False)

想请教大大佬,这个是什么原因导致onnx模型中ConvTranspose2d层weight的shape定义不一样。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions