Additional stoppers for ray tune
Project description
📝 Description
Ray tune is a tool for scalable hyperparameter tuning for machine learning applications. For intelligent hyperperameter optimization, trials that are performing inadequately should be stopped early. Part of this can be done by schedulers such as ASHA, but additional explicit stopping criteria are useful as well.
For example, a trial that has converged and is no longer producing better results, but is still outperforming all other running trials, will not be stopped by ASHA. Ray tune currently only provides three different stoppers: a plateau stopper, a maximum iterations stopper, and a timeout stopper.
This module aims to foster a greater variety of community maintained contributed stopping mechanisms.
📦 Installation
pip3 install rt_stoppers_contrib
🔥 Using it
For a full list of the stoppers offered, see the the documentation.
Using any of the stoppers is as easy as
from rt_stoppers_contrib import NoImprovementTrialStopper
tuner = tune.Tuner(
tune.Trainable,
tune_config=...,
run_config=air.RunConfig(stop=NoImprovementTrialStopper("my_metric"))
)
For more information, refer to the documentation.
🧰 Development setup
pip3 install pre-commit
cd <this repository>
pre-commit install
gitmoji -i
💖 Contributing
Your help is greatly appreciated! Suggestions, bug reports and feature requests are best opened as github issues. You are also very welcome to submit a pull request!
Bug reports and pull requests are credited with the help of the allcontributors bot.
⚖️ License
See LICENSE for more information. The logo is built from the official ray-tune logo together with this stop sign (CC 4.0).
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for rt_stoppers_contrib-1.1.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | c37d091301c007869dc3d7bcd95c5b0b48dacc13d30b1c215abeabfd984bc360 |
|
MD5 | afbe001b5adee8babed590acd02eacbf |
|
BLAKE2b-256 | f3e468b5fcab5876707b74809cdde23db1cd5b849ebf9c6a6856c9f3c990809d |
Hashes for rt_stoppers_contrib-1.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c6e78d9ab49a1e16d440c0e273c6fdb855be06e06eb0f17784791157bd6f21cf |
|
MD5 | 310600781517f528b472973c522c0484 |
|
BLAKE2b-256 | 399e2c8a81396b2befd4b25b714bc5a9eb0e6b23b1d5059ed52c60377620891c |