training_extensions
training_extensions copied to clipboard
Fix to work torch compile in detection
Summary
-
WIP: Test all models
-
Check List
- [x] detection
- [x] atss_mobilenetv2
- [x] atss_resnext101
- [x] rtmdet_tiny
- [x] ssd_mobilenetv2
- [x] yolov9_c
- [x] yolov9_m
- [x] yolov9_s
- [x] yolox_l
- [x] yolox_s
- [x] yolox_tiny
- [x] yolox_x
- [ ] rtdetr
- [ ] instance-segmentation
- [x] maskrcnn_r50
- [ ] maskrcnn_efficientnetb2b
- [ ] maskrcnn_swint
- [ ] maskrcnn_r50_tv
- [x] rtmdet_inst_tiny
- [ ] maskdino_r50
- [x] detection
-
rtdetr (det), maskrcnn_swint, maskrcnn_r50_tv, maskdino_r50 (ins-seg) are not working with compile for now
How to test
Checklist
- [ ] I have added unit tests to cover my changes.
- [ ] I have added integration tests to cover my changes.
- [ ] I have ran e2e tests and there is no issues.
- [ ] I have added the description of my changes into CHANGELOG in my target branch (e.g., CHANGELOG in develop).
- [ ] I have updated the documentation in my target branch accordingly (e.g., documentation in develop).
- [ ] I have linked related issues.
License
- [ ] I submit my code changes under the same Apache License that covers the project. Feel free to contact the maintainers if that's a concern.
- [ ] I have updated the license header for each file (see an example below).
# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0