Skip to content

Commit

Permalink
fix unit testing for conv layers and add correct mat multiplication i…
Browse files Browse the repository at this point in the history
…n conv layer
  • Loading branch information
mhajij committed Jun 16, 2023
1 parent d47dd17 commit c471ea2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion topomodelx/base/conv.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ def forward(self, x_source, neighborhood, x_target=None):
)

x_message = torch.mm(x_source, self.weight)
x_message_on_target = torch.mm(neighborhood, x_message)
x_message_on_target = torch.sparse.mm(neighborhood, x_message)

if self.aggr_norm:
neighborhood_size = torch.sum(neighborhood.to_dense(), dim=1)
Expand Down

0 comments on commit c471ea2

Please sign in to comment.