How to modify a Conv2d to Depthwise Separable Convolution?

21

self.pointwise.weight = torch.nn.Parameter(weight) is fine :grinning:, but it maybe changed a lot, as there is no scaling on the depthwise