Hello. when i run this code on cuda environment this error happend. would you please help me fixing it?

HennyJie / BrainGB

Officially Accepted to IEEE Transactions on Medical Imaging (TMI, IF: 11.037) - Special Issue on Geometric Deep Learning in Medical Imaging.

MIT License

169 stars 42 forks source link

Hello. when i run this code on cuda environment this error happend. would you please help me fixing it? #21

Closed wenjie-666 closed 10 months ago

wenjie-666 commented 10 months ago

ValueError: Encountered tensor with size 640000 in dimension 0, but expected size 3200. It seems that it's because of this code： z = conv(z, edge_index, edge_attr) in gat.py It doesn't seem as simple as the official website suggests：For a GAT-based setup, simply replace the GCN model initializations with GAT.

salmon1802 commented 4 months ago

I meet the same problem as you. How did you solve it？