在 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