Skip to content

CogDL v0.4.0

Choose a tag to compare

@cenyk1230 cenyk1230 released this 30 May 16:29
· 102 commits to master since this release
3206099

A new major release! 馃帀馃帀馃帀
The new v0.4.0 release refactors the data storage (from Data to Graph) and provides more fast operators to speed up GNN training. It also includes many self-supervised learning methods on graphs. BTW, we are glad to announce that we will give a tutorial on KDD 2021 in August. Please see this link for more details. 馃帀

New Features

  • Reformat Data Storage (from Data to Graph), edge_index from torch.Tensor to tuple(Tensor, Tensor). The inputs of each GNN are unified as one parameter graph.
  • #205 #210 #212 Add SDDMM operator
  • #234 Add multi-head SpMM operator and speed up edge_softmax.
  • #211 #222 Support distributed training

New Models

New Datasets

  • #226 Add ogbn-mag dataset

New Examples

  • #233 Add Simple-HGN model

Bug Fixes

  • #209 Fix STPGNN and heterogeneous task
  • #225 Fix TUDataset