• Tingyu Wang's avatar
    [Model] Add `dgl.nn.CuGraphSAGEConv` model (#5137) · bcf9923b
    Tingyu Wang authored
    
    
    * add CuGraphSAGEConv model
    
    * fix lint issues
    
    * update model to reflect changes in make_mfg_csr(), move max_in_degree to forward()
    
    * lintrunner
    
    * allow reset_parameters()
    
    * remove norm option, simplify test
    
    * allow full graph fallback option, add example
    
    * address comments
    
    * address reviews
    
    ---------
    Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
    bcf9923b
nn-pytorch.rst 3.99 KB