keras 中的層規范化重命名

Question

我正在嘗試更新生成從 TF1 到 TF2 的 model 檢查點的代碼。 但是，TF2 代碼將層歸一化重命名如下：

• ../../layer_normalization_4/../..
• ../../layer_normalization_1/../..
• ../../layer_normalization_5/../..

它將后綴 _n 添加到變量名稱。 我希望它只是../../layer_normalization/../..，沒有_n。 做一些研究，我注意到這可能是 Keras (tf2) 行為。

提前致謝

Answer 1

不幸的是，所有 Keras 層都必須具有唯一的名稱，例如：

import tensorflow as tf
from tensorflow.keras.layers import Dense, Input
from tensorflow.keras.models import Model

input = Input(shape=(50,))
model = Dense(200,activation='relu', name='dense1')(input)
model = Dense(200,activation='relu',name='dense1')(model)
output = Dense(2,activation='sigmoid')(model)
model = Model(input,output)

會報錯：

ValueError：名稱“dense1”在 model 中使用了 2 次。 所有圖層名稱都應該是唯一的。