If I withdraw a few layers from resnet50. Then, I wish I can use these withdrew layers and combine them into a new neural network.
Well, I wish I can make the following layers as an independent neural network. Like the following
class new_net(nn.Module):
model.conv1
model.bn1
When the new neural network is created. Then, I can use the new neural network like: model = new_net().