在做图训练的时候经常无法运行，使用mutag数据集gin模型有时候可以训练，有时候无法训练

[fusky]

🐛 Bug

To Reproduce

Steps to reproduce the behavior:

Expected behavior

Environment

• CogDL version:0.6
• OS (e.g., Linux):Linux
• Python version:3.9
• PyTorch version:1.12.1+cu113
• CUDA/cuDNN version (if applicable):11.4
• Any other relevant information:

Additional context