Я недавно обучил реализацию u-Net в Keras для сегментирования определенных объектов. Теперь я хотел бы загрузить веса из этой обученной модели и настроить несколько последних слоев в новом наборе данных.
Хотя это сложно, потому что исходная модель была обучена как multi_gpu_model (), и подход, который я попробовал, не поддается обработке. Я пытался сделать что-то вроде следующего:
for ii in model.layers[0:x]:
ii.trainable=False
Но это не сработает, потому что model.layers возвращает только входной слой и два лямбда-слоя, которые предположительно представляют экземпляр модели на каждой карте.
У кого-нибудь есть совет?