SIATune is an open-source deep learning model hyperparameter tuning toolbox especially for OpenMMLab's model frameworks such as mmdetection and mmsegmentation. In order to support job scheduling and resource management, SIATune adopts Ray and Ray.tune.
-
Fully support OpenMMLab models
We provide a unified model hyperparameter tuning toolbox for the codebases in OpenMMLab. The supported codebases are listed as below, and more will be added in the future.
-
Support hyperparameter search algorithms
We provide hyperparameter search algorithms such as below;
-
Schedule multiple experiments
Various scheduling techniques are supported to efficiently manage many experiments.
-
Distributed tuning system based on Ray
Hyperparameter tuning with multi-GPU training or large-scale job scheduling are managed by Ray's distributed compute framework.
Please refer to get_started.md for installation and getting started.
This project is released under the Apache 2.0 license.
If you use SIATune in your research, please use the following BibTeX entry.
@misc{na2022siatune,
author = {Younghwan Na and Hakjin Lee and Junhwa Song},
title = {SIATune},
howpublished = {\url{https://github.com/SIAnalytics/siatune}},
year = {2022}
}