8000 When converting the weight file to an onnx file, there are two identity nodes participating in the BN layer. Have you encountered this problem · Issue #98 · XuJiacong/PIDNet · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
8000
When converting the weight file to an onnx file, there are two identity nodes participating in the BN layer. Have you encountered this problem #98
Open
@Daniu-jiao

Description

@Daniu-jiao

This mistake in class PAPPM last shortcut moduel.

out = self.compression(torch.cat(x_list, 1)) + self.shortcut(x)

self.shortcut = nn.Sequential(
BatchNorm(inplanes, momentum=bn_mom),
nn.ReLU(inplace=True),
nn.Conv2d(inplanes, outplanes, kernel_size=1, bias=False), )

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

      0