ОШИБКА визуализатора ANN: слой не поддерживается для визуализации

Я относительно новичок в машинном обучении с помощью Python и Tensorflow/Keras. Теперь у меня есть работающая модель, и я хотел бы визуализировать свою сеть. Для этого существует библиотека python для визуализации искусственных нейронных сетей (ИНС): ANN Visualizer.

К сожалению, я получаю это сообщение об ошибке:

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
C:\Temp\ipykernel_17876\3726106579.py in 
     24 history = model.fit(trainx, trainy, batch_size=1, epochs=30, validation_data=(testsetx, testsety))
     25 
---> 26 ann_viz(model)

c:\Users\aalles\Anaconda3\lib\site-packages\ann_visualizer\visualize.py in ann_viz(model, view, filename, title)
    121                 c.node(str(n), label = "Image\n"+pxls[1]+" x"+pxls[2]+" pixels\n"+clrmap, fontcolor = "white");
    122             else:
--> 123                 raise ValueError("ANN Visualizer: Layer not supported for visualizing");
    124         for i in range(0, hidden_layers_nr):
    125             with g.subgraph(name = "cluster_"+str(i+1)) as c:

ValueError: ANN Visualizer: Layer not supported for visualizing

Мой метод построения модели выглядит так:

def build_ann(self, nLayers=4, nNeurons=64, LR_init=1e-2, LR_adapt=1e-4, LR_steps=1e5, regularizer=1e-2, dropout=0.3):
   
        model = Sequential()
        for runs in range(nLayers):
            return_seq = True if runs < nLayers-1 else False
            model.add(LSTM(int(Neurons[runs]), return_sequences=return_seq))  # , dropout=0.3
            model.add(Dropout(dropout))
        model.add(Dense(int(Neurons[-1])))
        model.add(Dense(len(self.targets), activation='sigmoid'))
        model.compile(loss='mean_squared_error', optimizer='Adam', metrics=['accuracy'])  
        model.build(input_shape=(1, self.maxlen, len(self.features)))
        model.summary()

        return model

Следовательно, объект будет создан:

model = ann.build_ann(nLayers=2, nNeurons=2)
history = model.fit(trainx, trainy, batch_size=1, epochs=30, validation_data=(testsetx, testsety))
ann_viz(model)

И появляется ошибка.

Почему нельзя создать визуализацию? В других примерах это было из-за слоя Flatter(). У меня нет такого слоя. Это цикл for?

Большое спасибо!