conv2 is wrong?

[AnthonyF333]

class PFLDInference(nn.Module): def init(self): super(PFLDInference, self).init()

    self.conv1 = nn.Conv2d(
        3, 64, kernel_size=3, stride=2, padding=1, bias=False)
    self.bn1 = nn.BatchNorm2d(64)
    self.relu = nn.ReLU(inplace=True)

    self.conv2 = nn.Conv2d(
        64, 64, kernel_size=3, stride=1, padding=1, bias=False)
    self.bn2 = nn.BatchNorm2d(64)
    self.relu = nn.ReLU(inplace=True)

@polarisZhao I think conv2 is a depth-wise convolution instead of a normal convolution, right?

[polarisZhao]

yes, you're right. Thanks a lot. I will fix it.