Hi everyone. May I ask if t’s possible to setup up a fixed number of output channels for models such as densenet/resnet on the different blocks? For instance I have a model and I would like to add densenet or resnet as a feature extractor model. But the problem that I’m facing at the moment is that for instance in the densenet/resnet models you have different blocks, either bottleneck or dense blocks, depending on the model. My goal it to take the output of each of the dense or bottleneck blocks but at the same time I would like to fix the the number of output channels to my predefined choice, is that possible? I’ve tried using num_init_features when initializing the models but that didn’t work I kept getting and error about unrecognizable keyword.