training_extensions
training_extensions copied to clipboard
Apply adaptive repeat data for all tasks and deprecate adaptive scheduling hook
Summary
- This PR brings the
AdaptiveRepeatDataHookthat will be used instead ofAdaptiveTrainingSchedulefor detection, s-seg, i-seg tasks. - Modified learning rate for two models including ATSS, Eff-b2b-Mask-RCNN that are sensitive to the learning rate of the early training phase.
- Renamed
AdaptiveSchedulingHooktoEvalBeforeRun.
This PR affects the performance, data_time, iter_time and e2e_time. And this is the summarized results
Experimental Datasets
- Detection : Fish, Vitens, Pothole, BCCD (Small)
- Instance Segmentation: Vitens-aeronomas-12, 100, wgisd-coco
- Semantic Segmentation: Kvasir-100, voc-200
How to test
Checklist
- [ ] I have added unit tests to cover my changes.
- [ ] I have added integration tests to cover my changes.
- [ ] I have added e2e tests for validation.
- [ ] 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) 2023 Intel Corporation
# SPDX-License-Identifier: Apache-2.0