PyTorch Forums
Intermediate Layers of AlexNet/VGG
vision
ptrblck
March 9, 2020, 1:07am
4
You could use forward hook as described in
this example
.
1 Like
show post in topic