有沒有辦法讀取/打印穩定基線中神經網絡隱藏層的激活？

Question

我正在從事的項目的核心要求是能夠讀取我使用穩定基線庫訓練的 PPO2 模型隱藏層中神經元的激活。 這是一個非常相關的問題。 我想按此處所示打印它們。 我最接近的是這樣做：

print(model.get_parameters())

這僅打印權重和偏差，但不打印預測時的激活。 我試圖編輯穩定基線庫的文件但無濟於事。 我也試過

print(model.policy)

這將返回 <class 'stable_baselines.common.policies.MlpPolicy'> 因為這僅指我正在使用的策略類型。 如果沒有辦法有效地做到這一點，是否可以輕松遷移我的簡單環境並使用另一個庫進行訓練？ 非常感謝我能得到的任何幫助/建議。

Answer 1

我認為最簡單的方法，不需要你重寫 pytorch 的方法，就是附加一個 forward hook 。 我想我在這里找到了完全適合您問題的代碼。 但是，您仍然需要定義一個自定義策略，在其中粘貼所有掛鈎附件，但在您的情況下這應該不是那么難。