Hi, I made a library to get the intermediate results, minimal modifications to your model (none for simple use cases) and if a module is called more than once all the results are saved. Check it out. Pytorch-Intermediate-Layer-Getter 32