AttributeError: 'str' object has no attribute 'get'

[chc8]

I receive an error about 'str' object has no attribute 'get':

from keras.models import Sequential
from keras.layers import Dense, Conv2D, Flatten, Dropout, MaxPooling2D

model = Sequential()

model.add(Conv2D(16, kernel_size=3, strides=3, activation='relu',
data_format="channels_last", input_shape=(129,1173,1)))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(32, kernel_size=3, strides=3, activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(64, kernel_size=3, strides=3, activation='relu'))
model.add(Flatten())
model.add(Dense(units=256, activation='relu'))
model.add(Dense(units=128, activation='relu'))
model.add(Dense(units=64, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(1))
model.compile(loss='mae', optimizer='adam')

model_draw = convert_drawer_model(model)
model_draw.save_fig("11.svg")

---

AttributeError Traceback (most recent call last)
in
----> 1 model_draw = convert_drawer_model(model)
2 model_draw.save_fig("11.svg")

~/Notebooks/convnet-drawer/keras_util.py in convert_drawer_model(model)
33 figure = Model(input_shape=_input_shape[1:])
34 for config in model.get_config():
---> 35 class_name = config.get("class_name", False)
36 class_config = config.get("config", False)
37 if class_name and class_config:

AttributeError: 'str' object has no attribute 'get'

[yu4u]

Thank you for your report!
I fixed this issue.
Please try it again with the current convent-drawer version.

81a840f

[chc8]

Thank you for fixing this. I really like this project. It helps me visualize the network design.