如何在权重衰减的 keras 中加载预训练模型

Question

我使用预训练的 Resnet 50 训练了一个模型，权重衰减为 10^-3。现在，当我尝试如下重新加载它时：

model = load_model("/content/drive/My Drive/prediction/tf_model.h5",custom_objects={'AdamWeightDecay': optimizer})
preproc = pickle.load(open("/content/drive/My Drive/prediction/tf_model.preproc",'rb'))

prediction = ktrain.get_predictor(model,preproc)

它显示以下错误：

NameError: name 'optimizer' is not defined

没有提到优化器，错误如下：

ValueError: Unknown optimizer: AdamWeightDecay. Please ensure this object is passed to the `custom_objects` argument. See https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for details.

谁能告诉我应该如何定义优化器。