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
Data
toGraph
),edge_index
fromtorch.Tensor
totuple(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