Migration TF2

Commande utile pour visualiser un modèle .pb

saved_model_cli show --dir chemin/du/modele --tag_set serve --signature_def serving_default

TF1

Les noms des noeuds sont du type tower_0/s2_t. On y accède via les nodes du GraphDef

TF2

Lors du parcours des nodes du GraphDef :

• Les noms des noeuds "inputs" sont du type serving_default_s2_t.
• Les noms des noeuds "outputs" sont du type StatefulPartitionedCall

A la place, il est plus pertinent de parcourir :

• bundle.GetSignatures().at("serving_default").inputs() qui est un mapping du type {"s2_t": node_info}
• bundle.GetSignatures().at("serving_default").outputs() qui est un mapping du type {"s2_t": node_info}

Ressources utiles :

• https://stackoverflow.com/a/64385815/13711499
• https://stackoverflow.com/questions/63181951/how-to-get-graph-or-graphdef-from-a-given-model
• https://stackoverflow.com/questions/58968918/accessing-input-and-output-tensors-of-a-tensorflow-2-0-savedmodel-via-the-c-api