Closed 943fansi closed 2 years ago
Describe the bug when use plot_model, raise ValueError
model define:
model define
inputs = tf.keras.layers.Input(shape=input_shape, name='input') tcn_out = TCN(nb_filters=64, kernel_size=3, nb_stacks=1, activation="LeakyReLU")(inputs) outputs = Dense(forecast_horizon * num_features, activation='linear')(tcn_out) outputs = tf.reshape(outputs, shape=(-1, forecast_horizon, num_features), name='ouput') model = tf.keras.Model(inputs=inputs, `outputs=outputs)`
tf.keras.utils.plot_model( model, to_file="TCN_model.png", show_shapes=True, show_dtype=True, show_layer_names=True, rankdir="TB", expand_nested=False, dpi=96, layer_range=None, show_layer_activations=True, )
tf.__version__: 2.8.0
tf.__version__
want to get model's png like https://github.com/philipperemy/keras-tcn/issues/112
Fixed in https://github.com/philipperemy/keras-tcn/commit/05cfa3da1469aed2499ab6b7aef9f37023c9833e
Keras TCN 3.4.4 pushed. Please update.
Describe the bug when use plot_model, raise ValueError
model define
:tf.__version__
: 2.8.0want to get model's png like https://github.com/philipperemy/keras-tcn/issues/112