mirror of
https://github.com/vale981/ray
synced 2025-03-09 04:46:38 -04:00
e89195bfb9
* 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>
58 lines
1.4 KiB
Text
58 lines
1.4 KiB
Text
autogluon.core==0.1.0
|
|
ax-platform==0.1.9; python_version < '3.7'
|
|
ax-platform==0.1.20; python_version >= '3.7'
|
|
bayesian-optimization==1.2.0
|
|
ConfigSpace==0.4.18
|
|
dask[complete]==2021.03.0; python_version < '3.7'
|
|
dask[complete]==2021.06.1; python_version >= '3.7'
|
|
dragonfly-opt==0.1.6
|
|
flaml==0.5.2
|
|
gluoncv==0.10.1.post0
|
|
gpy==1.10.0
|
|
gym[atari]==0.18.3
|
|
h5py==3.1.0; python_version < '3.7'
|
|
h5py==3.2.1; python_version >= '3.7'
|
|
hpbandster==0.7.4
|
|
HEBO==0.1.0
|
|
hyperopt==0.2.5
|
|
jupyter==1.0.0
|
|
keras==2.4.3
|
|
kubernetes==17.17.0
|
|
lightgbm==3.2.1
|
|
matplotlib==3.3.4; python_version < '3.7'
|
|
matplotlib==3.4.2; python_version >= '3.7'
|
|
mlflow==1.19.0
|
|
mxnet==1.8.0.post0
|
|
nevergrad==0.4.3.post3
|
|
optuna==2.8.0
|
|
pytest-remotedata==0.3.2
|
|
pytorch-lightning-bolts==0.2.5
|
|
pytorch-lightning==1.0.3
|
|
scikit-learn==0.24.2
|
|
scikit-optimize==0.8.1
|
|
sigopt==7.4.0
|
|
smart_open==5.0.0
|
|
tensorflow-probability==0.13.0
|
|
timm==0.4.5
|
|
|
|
|
|
torch==1.8.1;sys_platform=="darwin"
|
|
torchvision==0.9.1;sys_platform=="darwin"
|
|
|
|
# On non-OSX machines only install CPU version of torch and torchvision
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
torch==1.8.1+cpu;sys_platform!="darwin"
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
torchvision==0.9.1+cpu;sys_platform!="darwin"
|
|
|
|
|
|
transformers==4.9.1
|
|
wandb==0.10.29
|
|
xgboost==1.3.3
|
|
zoopt==0.4.1
|
|
|
|
# Dependencies for Hugging Face examples:
|
|
# `python/ray/util/sgd/v2/examples/transformers/transformers_example.py`
|
|
accelerate==0.3.0
|
|
datasets==1.11.0
|
|
sentencepiece==0.1.96
|