训练好网络，突然想获取网络中间的变量，怎么办

训练好网络，突然想获取网络中间的变量，是不是第一反应是改代码的返回值并且重写训练一遍？？

绝对不要，确实需要改变model代码的返回值，把需要的中间变量返回。但是不需要重新训练。

因为torch.load()加载的是权重文件.pth，但是仍然需要model的python文件才能运行，只要网络结构没变，返回值怎么改都没问题。