lbann icon indicating copy to clipboard operation
lbann copied to clipboard
verified publisher icon LLNL

FAST model

Open szaman19 opened this issue 5 years ago • 0 comments

LBANN implementation of the FAST model

Includes the following:

  • 3DCNN model for grid-structured data
  • SGCNN model for graph data
    • New implementation of GatedGraphConv with outer Edge network
    • New implementation of NNConv (edge-conditioned Graph NN)
    • New implementation of Graph Attention Layer
  • Fusion model: Combines intermediate layers of 3DCNN and SGCNN for regression

szaman19 avatar Oct 14 '20 19:10 szaman19