caffe在測試階段忘記設置變量

Question

在caffe中，如果我使自己的變量與一個圖層關聯，並且在訓練迭代過程中修改此變量，測試階段不知道對該變量的修改，它似乎將其初始化爲自己。訓練測試階段分享重量，但沒有別的。如何在培訓和測試階段共享其他變量。謝謝。

如： 說我在inner_product_layer.hpp聲明，

int dog = 5; 

和Backward_gpu（）我修改，

dog = 6; 

測試階段，認爲狗= 5。我想測試階段知道狗現在是6.

Answer 1

訓練和測試階段的可執行文件是不同的，因此，如果在訓練期間修改代碼中的變量，它不會反映在測試時間。您可以在訓練結束時將變量轉儲到輸出文件，並在測試時將該文件作爲參數讀取，同時初始化LayerSetUp函數中的圖層。文件名可以作爲圖層參數添加，您可以在其中檢查您是否處於測試階段，並根據您保存的文件設置參數。