Skip to content

v0.6.4

Compare
Choose a tag to compare
@bonevbs bonevbs released this 07 Dec 12:47
· 65 commits to main since this release
eab72f0
  • reworking distributed to allow for uneven split tensors, effectively removing the necessity of padding the transformed tensors
  • distributed SHT tests are now using unittest. Test extended to vector SHT versions. Tests are defined in torch_harmonics/distributed/distributed_tests.py
  • base pytorch container version bumped up to 23.11 in Dockerfile