...
for key, info in self.layers_info.items():
self.convs[key] = ConvKXBN(info, no_create=no_create, **kwargs)
self.convs[key].add_module('relu', nn.ReLU(inplace=True))
self.block_list.append(self.convs[key])
...
I want -
print(model)
HardConvK1KXK1(
(conv1): ConvKXBN(
(conv1): Conv2d(96, 24, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(bn1): BatchNorm2d(24, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(relu): ReLU(inplace=True)
)
)
But I’m getting -
HardConvK1KXK1()