Skip to content
This repository was archived by the owner on Jun 11, 2026. It is now read-only.

feat(GraphNorm): Support GraphNorm as option#29

Open
mmjb wants to merge 1 commit into
masterfrom
dev/mabrocks/graph_norm
Open

feat(GraphNorm): Support GraphNorm as option#29
mmjb wants to merge 1 commit into
masterfrom
dev/mabrocks/graph_norm

Conversation

@mmjb

@mmjb mmjb commented Sep 18, 2020

Copy link
Copy Markdown
Contributor

Implementation of Graph Norm (https://arxiv.org/pdf/2009.03294.pdf).

@mmjb mmjb requested a review from mallamanis September 20, 2020 08:40
Comment thread tf2_gnn/layers/gnn.py
self._dense_every_num_layers = params["dense_every_num_layers"]
self._residual_every_num_layers = params["residual_every_num_layers"]
self._use_inter_layer_layernorm = params["use_inter_layer_layernorm"]
self._use_graphnorm = params.get("use_graphnorm") or False

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Would it maybe make sense to add something like

assert not (self._use_graphnorm and self._use_inter_layer_layernorm), "Using layer normalization and graph normalization should not be used together."

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants