ptrblck
Is there a way to get the kernel?
For example if I have:
layer = Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
W = layer.weight
print(W.size())

The result will be:
torch.Size([128, 128, 3, 3])

But I want to save the kernel at the end of training, so only the 3x3 weights to be able to initialize the layer using this kernel in another network with requires_grad=False
How can I do this?