hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert "Revert "[RLlib](deps): Bump tensorflow from 2.5.0 to 2.6.0 in /python/requirements/rllib"" (#19352)

Browse source 
This reverts commit bde9e058da.
This commit is contained in:
Kai Fricke 2021-10-18 22:29:16 +01:00 • committed by GitHub
parent a081579f68
commit 6798bdbb5d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
python/ray/tune/examples/pbt_tune_cifar10_with_keras.py
View file

@ -16,11 +16,11 @@ import argparse
import numpy as np
import tensorflow as tf
from tensorflow.python.keras.datasets import cifar10
from tensorflow.python.keras.layers import Input, Dense, Dropout, Flatten
from tensorflow.python.keras.layers import Convolution2D, MaxPooling2D
from tensorflow.python.keras.models import Model, load_model
from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.datasets import cifar10
from tensorflow.keras.layers import Input, Dense, Dropout, Flatten
from tensorflow.keras.layers import Convolution2D, MaxPooling2D
from tensorflow.keras.models import Model, load_model
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from ray import tune
from ray.tune import Trainable

4
python/requirements/ml/requirements_dl.txt
View file

@ -1,8 +1,8 @@
# These requirements are used for the CI and CPU-only Docker images so we install CPU only versions of torch.
# For GPU Docker images, you should install requirements_ml_docker.txt instead.
tensorflow==2.5.0
tensorflow-probability==0.13.0
tensorflow==2.6.0
tensorflow-probability==0.14.0
torch==1.9.0;sys_platform=="darwin"
torchvision==0.10.0;sys_platform=="darwin"

2
python/requirements/ml/requirements_tune.txt
View file

@ -8,7 +8,7 @@ ConfigSpace==0.4.18
dask[complete]==2021.03.0; python_version < '3.7'
dask[complete]==2021.8.1; python_version >= '3.7'
dragonfly-opt==0.1.6
flaml==0.5.2
flaml==0.6.7
freezegun==1.1.0
gluoncv==0.10.1.post0
gpy==1.10.0