Hi all,
I’m looking to chop an exciting network up in parts so I could later chain different parts together, eg.
model = densenet121()
first_half = densenet121[:6]
second half = densenet121[6:]

However, this has not been so easy, I’ve tried splitting the model using model.parameters, model.modules and model.features, but I’ve not been able retain functionality when tied back together.
What is the proper way to copy or disregard one half of a model?