Hi;
I tried tweaking the forward pass and I think it was the easiest thing to do. However is there an nice way to truncate the model till the bottleneck layer, because I would now like to add an average pooling at the bottleneck layer to perform regression. Any ideas