search

microsoft / Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.
MIT License
2k stars 324 forks source link

preprocess_item()中入度求解问题 #192

Open JeremySun1224 opened 1 month ago

JeremySun1224 commented 1 month ago

wrapper.py line 53中的preprocess_item()中 item.in_degree = adj.long().sum(dim=1).view(-1) 如果是为了求入度, 是否应该是dim=0? (不过本例中都是无向图, 所以没有出错.)