Updated amg.py to allow batching of large mask sets and avoid over-sized torch.nonzero() calls#569
Open
MartinKlefas wants to merge 11 commits intofacebookresearch:mainfrom MartinKlefas:main
+55-1
Commits
Commits on Sep 21, 2023
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
