Plot TFRS model architecture

[lpiscusc]

I've been using tf.keras.utils.plot_model() to visualize Keras model architectures, which has been very helpful. However, I'm working with TFRS, and this function doesn't seem to work in this case. Is there an alternative way to visualize the TFRS model architecture?

Thanks

[Chm-vinicius]

Try tensorboard:

import datetime
%load_ext tensorboard

logdir = "logs/fit" + datetime.date.today().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=logdir)

final_model.fit(train.batch(500), epochs=10, callbacks=[tensorboard_callback])

See more on https://www.tensorflow.org/tensorboard/graphs