digger yu ce535945e6 fix: change ==NONE to is (#3923) | 1 year ago | |
---|---|---|
.. | ||
README.md | 4cf970e6bb Add codespell to pre-commit checks (#1717) | 2 years ago |
__init__.py | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 year ago |
base_tuner.py | ce535945e6 fix: change ==NONE to is (#3923) | 1 year ago |
cost_model.py | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 year ago |
index_based_tuner.py | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 year ago |
model_based_tuner.py | 3fb3cfdced fix some typo (#3675) | 1 year ago |
utils.py | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 year ago |
exps
is a list of experiment descriptions (dictionaries).
An experimentation description has a ds_config
field that stores the DeepSpeed configuration to be used in the experiment.
A tuner is based on BaseTuner and at least implements the next_batch
method. It can implement a different tune
method from the BaseTuner's.
class NewTuner(BaseTuner):
def __init__(self, exps: list, resource_manager):
super(NewTuner, self).__init__(exps, resource_manager)
def next_batch(self, sample_size=1):
pass
def tune(self): # if it differs from BaseTuner
pass