Я хочу сохранить свою конфигурацию модели CNN (размеры ядра, активации, фильтры и т. Д.) В текстовый файл.Используя «сводку», я получаю только ввод, вывод и параметры слоев, но мне нужно больше информации.
Я попробовал следующие функции:
# large dictionary with all information, but have a lot of noise info
config = model.get_config()
# same dictionary, but converted to string
summaryJson = str(model.to_json())
Все эти решения не дают мнесамые важные параметры.Так что я нашел это решение, которое, кажется, дает все, что мне нужно, но оно не работает:
from keras_diagram import ascii
summary = asc(model)
Но оно дает мне следующую ошибку:
AttributeError: 'Activation' object has no attribute 'inbound_nodes'
Это мой последнийслои:
...
hid = Conv2D(128, kernel_size=5, strides=1, padding='same')(hid)
hid = BatchNormalization(momentum=0.9)(hid)
hid = LeakyReLU(alpha=0.1)(hid)
hid = Conv2D(3, kernel_size=5, strides=1, padding="same")(hid)
out = Activation("tanh")(hid)
model = Model(input_layer, out)
summary = ascii(model)
Вы знаете, что делать?