PyTorch Forums
How to get output of layers?
vision
ptrblck
April 6, 2020, 3:31am
2
You could use forward hook to get the layers outputs as described
here
.
2 Likes
show post in topic