如何保存經過訓練的神經網絡的權重和偏差

Question

我使用 tensorflow 構建了神經網絡。

在訓練它並獲得最低成本之后，我需要將權重和偏差保存為矩陣，以便在例如 Matlab 中使用。 我怎么能這樣做？

當我在完成訓練后執行以下代碼以保存權重時：

from scipy.io import savemat
savemat("end_Weight1.mat", weights )

我可以保存它，但我得到空矩陣，這意味着矩陣只包含四個變量但沒有任何值。 我在矩陣中沒有得到任何值。

我認為在保存它們之前需要更新權重或類似的東西？

Answer 1

您可以使用

all_variables=tf.global_variables()

在每個批次上從創建的圖表中獲取變量，然后保存它們。

注意：此方法返回所有變量，因此您可以過濾列表。例如，您可以根據需要按偏差或權重對其進行過濾。