加载mod后在Keras中使用自定义优化器

Question

我想在keras中使用AdamW optimizer进行NN培训，根据： https://github.com/GLambard/AdamW_Keras

我导入它，代码看起来像广告一样工作。不幸的是，当我加载保存的检查点时，我收到一个错误：

file "ResUNET.py", line 209, in <module>
'peteLoss': mc.peteLoss, 'iou_one_Metric': mc.iou_one_Metric, 'adamw': adamw})
File "/home/pete/anaconda3/lib/python3.6/site-packages/keras/engine/saving.py", line 276, in load_model
custom_objects=custom_objects)
File "/home/pete/anaconda3/lib/python3.6/site-packages/keras/optimizers.py", line 760, in deserialize
printable_module_name='optimizer')
File "/home/pete/anaconda3/lib/python3.6/site-packages/keras/utils/generic_utils.py", line 138, in deserialize_keras_object
': ' + class_name)
ValueError: Unknown optimizer: AdamW

我确实根据以下条件传递自定义对象：

^{pr2}$

这对损失等有效，但对优化器无效。在