• Lingfan Yu's avatar
    [Model] Update GAT model code (#622) · 74e13eea
    Lingfan Yu authored
    * fix gat code to use latest edge softmax module
    
    * avoid transpose
    
    * update README
    
    * use edge_softmax op
    
    * mxnet edge softmax op
    
    * mxnet gat
    
    * update README
    
    * fix unittest
    
    * fix ci
    
    * fix mxnet nn test; relax criteria for prod reducer
    74e13eea
__init__.py 1.57 KB