search

Lin-Yijie / Graph-Matching-Networks

PyTorch implementation of Graph Matching Networks, e.g., Graph Matching with Bi-level Noisy Correspondence (COMMON, ICCV 2023), Graph Matching Networks for Learning the Similarity of Graph Structured Objects (GMN, ICML 2019).
Other
271 stars 55 forks source link

GraphEncoder MLP edge #2

Closed tathagatv closed 2 years ago

tathagatv commented 2 years ago

In line 46 of graphembeddingnetwork.py, the iteration should be done with limit len(self._edge_hidden_sizes) while iterating over self._edge_hidden_sizes. Is using len(self._node_hidden_sizes) there a mistake?

Lin-Yijie commented 2 years ago

Yes, it is a mistake! I have fixed it right now. Thanks~