1.4.2
版本发布时间: 2020-02-18 23:26:06
pyg-team/pytorch_geometric最新发布版本:2.5.3(2024-04-19 19:37:44)
Minor Changes
- There are now Python wheels available for
torch-scatter
andtorch-sparse
which should make the installation procedure much more user-friendly. Simply run
pip install torch-scatter==latest+${CUDA} torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-geometric
where ${CUDA}
should be replaced by either cpu
, cu92
, cu100
or cu101
depending on your PyTorch installation.
-
torch-cluster
is now an optional dependency. All methods that rely ontorch-cluster
will result in an error requesting you to installtorch-cluster
. -
torch_geometric.data.Dataset
can now also be indexed and shuffled:
dataset.shuffle()[:50]
Bugfixes
- Fixed a bug that prevented the user from saving
MessagePassing
modules. - Fixed a bug in
RGCNConv
when usingroot_weight=False
.