Yes, according to the architecture of MobileNet V2 6 you can slice off feature layer-wise because model.features is defined as self.features = nn.Sequential(*features) and nn.Sequential behaves like python list of all layers (so can be sliced). But for other models, you need to check its implementation. You check the model implementation here 17, by finding the right model and clicking on [SOURCE] .