Skip to content

Commit

Permalink
fix readme
Browse files Browse the repository at this point in the history
  • Loading branch information
lucidrains committed Mar 28, 2021
1 parent e7a8b8c commit a8dfbd2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ mask = torch.ones_like(feats).bool() # (1, 1024)
feats_out, coors_out = net(feats, coors, mask = mask) # (1, 1024, 32), (1, 1024, 3)
```

Only attend to sparse neighbors, given to the network as an adjacency matrix. `num_nearest_neighbors` will need to be set to the maximum number of neighbors to cover all the neighbors as specified in the matrix.
Only attend to sparse neighbors, given to the network as an adjacency matrix.

```python
import torch
Expand Down

0 comments on commit a8dfbd2

Please sign in to comment.