tensorflow / tensorflow/model-optimization

How to disable change the output tensor name?

Open
#28 2 comments 0 reactions 1 assignee View on GitHub

@liyunlu0618 is already working on this.

Since Apr 18, 2021.

feature request technique:pruning
Dominant language
Python
Stars
1.6k
Forks
349
Avg merge
3d 2h
Merged PRs (30d)
1

Description

My model has multiple outputs when using model-optimization, the outputs node name has been changed like this:

982/982 [==============================] - 434s 442ms/step - loss: 17.4818 - prune_low_magnitude_l1_loss: 3.6453 - prune_low_magnitude_l2_loss: 10.2949 - prune_low_magnitude_l1_p: 0.8551 - prune_low_magnitude_l1_r: 0.6556 - prune_low_magnitude_l2_p: 0.7639 - prune_low_magnitude_l2_r: 0.4423 - val_loss: 16.9125 - val_prune_low_magnitude_l1_loss: 3.4677 - val_prune_low_magnitude_l2_loss: 9.9504 - val_prune_low_magnitude_l1_p: 0.8921 - val_prune_low_magnitude_l1_r: 0.7815 - val_prune_low_magnitude_l2_p: 0.8306 - val_prune_low_magnitude_l2_r: 0.5363

how to remove the prefix 'prune_low_magnitude' ?

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.