the code about collect activations while inferencing

[mli-tian]

Hello! I really appreciate your work and try to understand the work, but I can't find the specific code to collect activations from each layer while performing inference. Can you tell me in which file? Thank you very much!

[LiuJinzhe-Keepgoing]

哈罗！我真的很感谢你的工作，并试图理解这项工作，但我在执行推理时找不到从每个层收集激活的具体代码。你能告诉我在哪个文件里吗？非常感谢！

I have encountered the same confusion. Have you found a solution? Can you share the code?