lbann
lbann copied to clipboard
Mesh-Graph Network Example
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
To do:
- [ ] Add normalizer of initial input
- [ ] Add an integration test with synthetic data
@szaman19 for normalizer you can use batchnorm afaiu