Hi, how can I copy specific layer structures and weights to a new model? I need to use Resnet but with additional layers and operations. What would be the best practice?
here’s my sample code
class Encoder(nn.Module):
def __init__(self, args1, args2):
super().__init__()
resnet = models.resnet50(pretrained=True)
# conv1, bn, relu, maxpool
self.stem = resnet.stem
# resnet block 1,2
self.layer1 = resnet.layer1
self.layer2 = resnet.layer2
# some new layers
self.layer3 = MultiInputBlock(args)
def forward(self, x, y):
x = self.stem(x)
x = self.layer1(x)
x = self.layer2(x)
if some_conditions:
x = self.layer3(x, y)
return x