可以看到InMemoryDataset 对CPU更加友好
https://pytorch-geometric.readthedocs.io/en/latest/modules/data.html#pytorch-lightning-wrappers
PyG两个data Datsaset v.s. InMemoryDataset
最新推荐文章于 2024-06-30 17:13:46 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
可以看到InMemoryDataset 对CPU更加友好
https://pytorch-geometric.readthedocs.io/en/latest/modules/data.html#pytorch-lightning-wrappers