tensorflow / tensorflow/recommenders

[Question] How to properly visualize the model structure as well as save model?

Open
#447 7 comments 0 reactions 1 assignee View on GitHub

@maciejkula is already working on this.

Since Jun 3, 2022.

question
Dominant language
Python
Stars
2k
Forks
300
PR merge metrics
No merged PRs in 30d

Description

I tried tf.keras.utils.plot_model

But it seems not to work.. only print the model name, is it because we did not use Functional API here?

Also for saving the model, model.save() works but later the weights and summary can not be called even if I do loaded_model(inputs) first.

I am also a little bit confused of Model vs. Layer class. I see all tutorials here use Model class for both query and candidate tower, but -> https://www.tensorflow.org/guide/keras/custom_layers_and_models#the_model_class . It says if you do not need to call fit (which is the case for query and candidate tower), consider use layer class

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.