ray/python/requirements/ml/requirements_tune.txt

48 lines
1.1 KiB
Text
Raw Normal View History

-r requirements_dl.txt
autogluon.core==0.1.0
ax-platform==0.1.9; python_version < '3.7'
ax-platform==0.2.1; python_version >= '3.7'
bayesian-optimization==1.2.0
ConfigSpace==0.4.18
dragonfly-opt==0.1.6
flaml==0.6.7
freezegun==1.1.0
gluoncv==0.10.1.post0
gpy==1.10.0
[RLlib] Upgrade gym version to 0.21 and deprecate pendulum-v0. (#19535) * Fix QMix, SAC, and MADDPA too. * Unpin gym and deprecate pendulum v0 Many tests in rllib depended on pendulum v0, however in gym 0.21, pendulum v0 was deprecated in favor of pendulum v1. This may change reward thresholds, so will have to potentially rerun all of the pendulum v1 benchmarks, or use another environment in favor. The same applies to frozen lake v0 and frozen lake v1 Lastly, all of the RLlib tests and have been moved to python 3.7 * Add gym installation based on python version. Pin python<= 3.6 to gym 0.19 due to install issues with atari roms in gym 0.20 * Reformatting * Fixing tests * Move atari-py install conditional to req.txt * migrate to new ale install method * Fix QMix, SAC, and MADDPA too. * Unpin gym and deprecate pendulum v0 Many tests in rllib depended on pendulum v0, however in gym 0.21, pendulum v0 was deprecated in favor of pendulum v1. This may change reward thresholds, so will have to potentially rerun all of the pendulum v1 benchmarks, or use another environment in favor. The same applies to frozen lake v0 and frozen lake v1 Lastly, all of the RLlib tests and have been moved to python 3.7 * Add gym installation based on python version. Pin python<= 3.6 to gym 0.19 due to install issues with atari roms in gym 0.20 Move atari-py install conditional to req.txt migrate to new ale install method Make parametric_actions_cartpole return float32 actions/obs Adding type conversions if obs/actions don't match space Add utils to make elements match gym space dtypes Co-authored-by: Jun Gong <jungong@anyscale.com> Co-authored-by: sven1977 <svenmika1977@gmail.com>
2021-11-03 08:24:00 -07:00
autorom[accept-rom-license]
gym[atari]>=0.21.0; python_version >= '3.7'
gym[atari]==0.19.0; python_version < '3.7'
h5py==3.1.0
hpbandster==0.7.4
pymoo<0.5.0 # this is a HEBO dependency, remove after https://github.com/huawei-noah/noah-research/issues/41 is fixed
HEBO==0.1.0
hyperopt==0.2.5
jupyter==1.0.0
kubernetes==17.17.0
lightgbm==3.2.1
matplotlib==3.3.4; python_version < '3.7'
matplotlib==3.4.3; python_version >= '3.7'
mlflow==1.21.0
mxnet==1.8.0.post0
nevergrad==0.4.3.post7
optuna==2.9.1
pytest-remotedata==0.3.2
lightning-bolts==0.4.0
pytorch-lightning==1.4.9
scikit-learn==0.24.2
scikit-optimize==0.8.1
sigopt==7.5.0
smart_open==5.1.0
timm==0.4.5
transformers==4.10.0
wandb==0.12.5
xgboost==1.3.3
zoopt==0.4.1
[SGD] add SGDv2 Trainer prototype implementation (#17440) * wip * formatting * increase timeouts * wip * address comments * comments * fix * address comments * Update python/ray/util/sgd/v2/worker_group.py Co-authored-by: Richard Liaw <rliaw@berkeley.edu> * Update python/ray/util/sgd/v2/worker_group.py Co-authored-by: Richard Liaw <rliaw@berkeley.edu> * address comments * formatting * fix * wip * finish * fix * formatting * remove reporting * split TorchBackend * fix tests * address comments * add file * more fixes * remove default value * update run method doc * add comment * minor doc fixes * lint * add args to BaseWorker.execute * address comments * remove extra parentheses * properly instantiate backend * fix some of the tests * fix torch setup * fix type hint * [SGD] add SGDv2 Trainer prototype implementation * add fashion mnist test * add HuggingFace example * format * formatting * address comment * address comments * update comment * Update python/ray/util/sgd/v2/examples/transformers/cluster.yaml Co-authored-by: Antoni Baum <antoni.baum@protonmail.com> * update huggingface transformers * update hugging face transformers * fix shutdown on worker failure * Update python/requirements/tune/requirements_tune.txt * Update python/requirements/tune/requirements_tune.txt * Update python/requirements/tune/requirements_tune.txt * Update python/requirements/tune/requirements_tune.txt * address comment and fix test Co-authored-by: Amog Kamsetty <amogkamsetty@yahoo.com> Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com> Co-authored-by: Richard Liaw <rliaw@berkeley.edu> Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
2021-08-02 15:27:42 -07:00
# Dependencies for Hugging Face examples:
# `python/ray/train/examples/transformers/transformers_example.py`
accelerate==0.5.1
datasets==1.14.0
[SGD] add SGDv2 Trainer prototype implementation (#17440) * wip * formatting * increase timeouts * wip * address comments * comments * fix * address comments * Update python/ray/util/sgd/v2/worker_group.py Co-authored-by: Richard Liaw <rliaw@berkeley.edu> * Update python/ray/util/sgd/v2/worker_group.py Co-authored-by: Richard Liaw <rliaw@berkeley.edu> * address comments * formatting * fix * wip * finish * fix * formatting * remove reporting * split TorchBackend * fix tests * address comments * add file * more fixes * remove default value * update run method doc * add comment * minor doc fixes * lint * add args to BaseWorker.execute * address comments * remove extra parentheses * properly instantiate backend * fix some of the tests * fix torch setup * fix type hint * [SGD] add SGDv2 Trainer prototype implementation * add fashion mnist test * add HuggingFace example * format * formatting * address comment * address comments * update comment * Update python/ray/util/sgd/v2/examples/transformers/cluster.yaml Co-authored-by: Antoni Baum <antoni.baum@protonmail.com> * update huggingface transformers * update hugging face transformers * fix shutdown on worker failure * Update python/requirements/tune/requirements_tune.txt * Update python/requirements/tune/requirements_tune.txt * Update python/requirements/tune/requirements_tune.txt * Update python/requirements/tune/requirements_tune.txt * address comment and fix test Co-authored-by: Amog Kamsetty <amogkamsetty@yahoo.com> Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com> Co-authored-by: Richard Liaw <rliaw@berkeley.edu> Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
2021-08-02 15:27:42 -07:00
sentencepiece==0.1.96