lbann icon indicating copy to clipboard operation
lbann copied to clipboard

Mesh-Graph Network Example

Open szaman19 opened this issue 3 years ago • 2 comments

A partially completed data parallel implementation of Mesh Graph Network

Partially completed as the Normalizer operation in the original is currently unsupported, but potentially could be simply added via an external layer. It is similar to Batch Norm but with some slight changes.

Pinging: @bvanessen @tbennun @ndryden

szaman19 avatar Feb 02 '23 07:02 szaman19

To do:

  • [ ] Add normalizer of initial input
  • [ ] Add an integration test with synthetic data

szaman19 avatar Feb 22 '23 22:02 szaman19

@szaman19 for normalizer you can use batchnorm afaiu

tbennun avatar Feb 23 '23 22:02 tbennun