Draft: Testing using nested tensors
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