[autoscaler] rsync cluster (#4785)

This commit is contained in:
Richard Liaw
2019-05-16 23:11:06 -07:00
committed by GitHub
parent ffe61fcc70
commit 88b45a53d6
4 changed files with 42 additions and 24 deletions
@@ -49,6 +49,11 @@ parser.add_argument(
action="store_true",
default=False,
help="disables CUDA training")
parser.add_argument(
"--redis-address",
default=None,
type=str,
help="The Redis address of the cluster.")
parser.add_argument(
"--seed",
type=int,
@@ -173,7 +178,7 @@ if __name__ == "__main__":
from ray import tune
from ray.tune.schedulers import HyperBandScheduler
ray.init()
ray.init(redis_address=args.redis_address)
sched = HyperBandScheduler(
time_attr="training_iteration", reward_attr="neg_mean_loss")
tune.run(