Thank you for the prompt reply.
My Pytorch version is 1.7.1 and my Pytorch geometric version is 1.6.3. I shall try to update them to the latest versions and try again.

BE01A2C4A5D44938A5AAFFA26E1EF9F4.png708×1 83 Bytes