ptrblck:

class Identity(nn.Module): def init(self): super(Identity, self).init() def forward(self, x): return x


Hi @ptrblck,
If we want to delete some sequenced layers in pretrained model, How could we do? for example in renet assume that we just want first three layers with fixed weights and omit the rest, I should put Identity for all layers I do not want? not any other way?