Skip to content

Override gnn_model_base trainable_variables#341

Merged
boomanaiden154 merged 2 commits into
mainfrom
users/boomanaiden154/override-gnn_model_base-trainable_variables
May 19, 2025
Merged

Override gnn_model_base trainable_variables#341
boomanaiden154 merged 2 commits into
mainfrom
users/boomanaiden154/override-gnn_model_base-trainable_variables

Conversation

@boomanaiden154

Copy link
Copy Markdown
Collaborator

This patch provides a custom implementation of trainable_variables in
gnn_model_base. Theoretically this should have been made unnecessary by
#337, but the interanl version of Tensorflow refuses to recurse into
the modules inside of the GraphNetworkLayer classes. This patch fixes
that by just returning the values regardless.

Created using spr 1.3.4
boomanaiden154 added a commit to boomanaiden154/gematria that referenced this pull request May 17, 2025
This patch provides a custom implementation of trainable_variables in
gnn_model_base. Theoretically this should have been made unnecessary by
\google#337, but the interanl version of Tensorflow refuses to recurse into
the modules inside of the GraphNetworkLayer classes. This patch fixes
that by just returning the values regardless.

Pull Request: google#341
Comment thread gematria/granite/python/gnn_model_base.py Outdated
Comment thread gematria/granite/python/gnn_model_base.py Outdated
Created using spr 1.3.4
@boomanaiden154 boomanaiden154 merged commit cdf8b1f into main May 19, 2025
@boomanaiden154 boomanaiden154 deleted the users/boomanaiden154/override-gnn_model_base-trainable_variables branch May 19, 2025 16:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants