Skip to content

Draft: Testing using nested tensors

Jackson Barr requested to merge jabarr/salt:nested into main

This MR details some tests with using nested tensors to avoid using padding. Nested tensors are still new and as such are missing a number of features which has made replacing the padded versions of some operations with nested equivalents difficult and likely slower

ToDo for tests

  • Run training loop without auxiliary tasks
  • Implement nested tensors for track origin and vertexing
  • Re-pad outputs when writing to a file for inference (?)
  • Check closure between nested and padded versions of GN2
  • Perform timing tests
Edited by Jackson Barr

Merge request reports