def regularization(layer, layer_in, layerId): l1 = layer['params']['l1'] l2 = layer['params']['l2'] out = {layerId: ActivityRegularization(l1=l1, l2=l2)(*layer_in)} return out