在 CUDA 上,SetTransformerAggregation 在 eval 模式下会引发错误(但不在 CPU 上)

作者: delkind-dnsf创建于 2025年4月16日更新于 2026年7月30日
标签bug

This is a weird one. I've made a simple modification of the GraphSAGE PPI example which replaces the aggregation method with the SetTransformerAggregation. We only make 2 simple changes to the example script.

内容来源: pyg-team/pytorch_geometric