Skip to content

Allow mean aggregation for InteractionGNN2

Jay Chan requested to merge jay_scatter_mean_for_interactionGNN2 into dev

Add the option to use mean as the aggregation for InteractionGNN2. One can specify in the config with the keyword aggr. The default (if not specified) is add.

Merge request reports