search

RedaOps / ann-visualizer

A python library for visualizing Artificial Neural Networks (ANN)
MIT License
1.24k stars 216 forks source link

ValueError #22

Open pku-H2R opened 5 years ago

pku-H2R commented 5 years ago

File "D:\Anaconda\lib\site-packages\ann_visualizer\visualize.py", line 42, in ann_viz input_layer = int(str(layer.input_shape).split(",")[1][1:-1]); ValueError: invalid literal for int() with base 10: ''

pikaliov commented 5 years ago

I have same error (using Keras API)

input_ = Input(batch_shape=(None, 1, dim_in), name='main_input')
out = Bidirectional(CuDNNGRU(256, return_sequences=True), name='BLSTM-256')(input_)
out = Bidirectional(CuDNNGRU(128, return_sequences=True), name='BLSTM-128')(out)
out = Bidirectional(CuDNNGRU(64, return_sequences=True), name='BLSTM-64')(out)
out = Bidirectional(CuDNNGRU(1, return_sequences=True), name='BLSTM-1')(out)
out = TimeDistributed(Dense(dim_out, activation='linear',), name='main_out')(out)
m = Model(inputs=[input_], outputs=[out])

X_train = [1900, 1, 6]
Y_train = [1900, 1, 4]
pikaliov commented 5 years ago

m_stack_blstm_mask