Closed JPXKQX closed 1 week ago
All modified and coverable lines are covered by tests :white_check_mark:
Project coverage is 99.85%. Comparing base (
deba13a
) to head (9e40145
). Report is 2 commits behind head on develop.
:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
This PR introduces the
NamedNodesAttributes(nn.Module)
class, designed to enhance the management of node attributes within graph neural networks. Key features include:grid_size
attribute, indicating the total number of nodes.in_channels_src/dst
, which represents the dimensionality of the node attribute tensor.The main improvement is the ability to handle an arbitrary number of nodes. This development will facilitate extending AnemoiModelEncProcDec to other use cases, such as hierarchical models or multi-encoder/decoder.