🚨🚨🚨 fix(Mask2Former): torch export 🚨🚨🚨 (#34393)
* fix(Mask2Former): torch export
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* revert level_start_index and create a level_start_index_list
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* Add a comment to explain the level_start_index_list
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* Address comment
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* add torch.export.export test
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* rename arg
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* remove spatial_shapes
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* Use the version check from pytorch_utils
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
* [run_slow] mask2former
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>
---------
Signed-off-by: Phillip Kuznetsov <philkuz@gimletlabs.ai>