对keras预训练模型添加l2正则化

最新推荐文章于 2022-03-01 15:55:41 发布

愤怒的勇

最新推荐文章于 2022-03-01 15:55:41 发布

阅读量956

点赞数

文章标签： tensorflow 神经网络机器学习

本文链接：https://blog.csdn.net/qq_27039891/article/details/104318359

版权

#以densenet为例
model = keras.applications.densenet.DenseNet121(input_shape=(224,224,3),
                                         weights=None,classes=100)
#对每个层添加l2正则化
for layer in model.layers:
    if hasattr(layer,'kernel_regularizer'):
    	layer.kernel_regularizer = keras.regularizers.l2(0.001)