Tutorial on net surgery?

Question

I'm trying to replace the "fc1000" layer in a NetModel, what am I missing here:

net = NetModel["ResNet-152 Trained on ImageNet Competition Data"]
NetReplacePart[net, {
     "fc1000" -> LinearLayer[2],
     "prob" -> SoftmaxLayer[],
     "Output" -> NetDecoder[{"Class", {"A", "B"}}]
 }]

I would love it if someone could write a nice tutorial on real-world applications of inserting/tweaking/moving layers for partial retraining.

Answer 1

OK, so NetReplacePart won't work in this case, you need a NetTake:

res = NetModel["ResNet-152 Trained on ImageNet Competition Data"]
net = NetChain[{NetTake[res, {1, "flatten_0"}], LinearLayer[2], 
   SoftmaxLayer[]}, "Output" -> NetDecoder[{"Class", {"A", "B"}}]]