search

tensorflow / gnn

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.
Apache License 2.0
1.3k stars 167 forks source link

Warning when saving model #816

Closed ydalmia closed 5 days ago

ydalmia commented 5 days ago

I'm encountering the following warning:

"UserWarning: Encoding a StructuredValue with type tensorflow_gnn.GraphTensorSpec; loading this StructuredValue will require that this type be imported and registered."

My model primarily uses gatconv2 and the warning occurs when I try to save the model to disk using model.save(f"savedmodels/model{epoch}").

Does anyone know what's going on here?

ydalmia commented 5 days ago

Sorry, I didn't see that there is a section on saving models. Here's the link in case anyone else faces this issue:

https://github.com/tensorflow/gnn/blob/main/tensorflow_gnn/docs/guide/model_saving.md