In the torch vision https://github.com/pytorch/vision/blob/master/torchvision/models/densenet.py#L195 5
for i, num_layers in enumerate(block_config):    
      block = _DenseBlock(num_layers=num_layers, num_input_features=num_features, bn_size=bn_size, growth_rate=growth_rate, drop_rate=drop_rate) 
      self.features.add_module('denseblock%d' % (i + 1), block) 

I want to apply a classifier for the block ( likes deep supervision) . How can I implement it? Thanks