(self)
| 114 | return eq_k, eq_b |
| 115 | |
| 116 | def merge_kernel(self): |
| 117 | eq_k, eq_b = self.get_equivalent_kernel_bias() |
| 118 | self.lkb_reparam = get_conv2d(in_channels=self.lkb_origin.conv.in_channels, |
| 119 | out_channels=self.lkb_origin.conv.out_channels, |
| 120 | kernel_size=self.lkb_origin.conv.kernel_size, stride=self.lkb_origin.conv.stride, |
| 121 | padding=self.lkb_origin.conv.padding, dilation=self.lkb_origin.conv.dilation, |
| 122 | groups=self.lkb_origin.conv.groups, bias=True) |
| 123 | self.lkb_reparam.weight.data = eq_k |
| 124 | self.lkb_reparam.bias.data = eq_b |
| 125 | self.__delattr__('lkb_origin') |
| 126 | if hasattr(self, 'small_conv'): |
| 127 | self.__delattr__('small_conv') |
| 128 | |
| 129 | |
| 130 | class ConvFFN(nn.Module): |
no test coverage detected