CogDL v0.4.0
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
DatatoGraph),edge_indexfromtorch.Tensortotuple(Tensor, Tensor). The inputs of each GNN are unified as one parametergraph. - #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