Feature Vector Extraction from Densenet121

2

If overriding the forward method is too cumbersome, you could use forward hooks instead as described here.