Hi @partially_observed and @EvanZ
If you’d like to extract input and/or output from some modules in your model, you can easily achieve it by using ForwardHookManager in torchdistill 11.
Here is an example notebook 30 to demonstrate the feature.