Add layers on pretrained model

[Nikronic (Nikan Doosti)]

Hi,

I think this post might help you:

Load only a part of the network with pretrained weights

Hi, For this, there are different approaches but personally, I would create a class on top of previously defined model which you have weights for. Then add any other layers as another sequential to the new defined model. Something like this: pretrained = torchvision.models.alexnet(pretrained=True) class MyAlexNet(nn.Module): def __init__(self, my_pretrained_model): super(MyAlexNet, self).__init__() self.pretrained = my_pretrained_model self.my_new_layers = nn.Sequen…

Bests