@my3bikaht
My actual use case is a bit different, I’m using large, pretrained EfficientNets there and I want to access certain layers, so using the intermediate values directly works but is not too flexible.
@anantguptadbl
Thanks! I really missed the detach() part, thats where my problem was!