search

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
179 stars 43 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 11 months ago

wenjie-666 commented 11 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 5 months ago

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