from bigdl.keras.converter import WeightsConverter
def set_weights_per_layer(kmodel, zmodel, layer_name):
    klayer = kmodel.get_layer(layer_name)
    klayer_weights = klayer.get_weights()
    zlayer_weights = WeightsConverter.to_bigdl_weights(klayer, klayer_weights)
    zlayer = [l for l in zmodel.layers if l.name() == layer_name][0] # assert the result length is 1
    zlayer.set_weights(zlayer_weights)