PyTorch Forums
Extracting speccific layers' output from a module list
ptrblck
August 10, 2022, 6:54pm
2
You can use forward hooks as described
here
and register them to the 4th and 10th layers.
show post in topic