Hi,
You need to change the forward method of your network. So that it uses the new fully connected layer.


How Can ?


What you printed above is just the list of all the parameters of your Module.
To use these parameters, you need to make sure they are used in the forward function of this module.
For example, here 1 is the original forward method for a resnet as it is defined in torchvision. To use a new layer, you would need to subclass this one and change the forward method to use what you want to use.


For example?


Thanks.