In that case I’ll need to change self.layers and the forward section, sure. I was just wondering in regard to changes in load_model.
Good to know that the same trick would work even if a new layer is not defined in the pretrained model. So basically I can add whatever I want and assign it to the relevant weights. Things are clearer now. Thank you