mirror of
https://github.com/wassname/ray.git
synced 2026-07-16 11:21:10 +08:00
[autoscaler] rsync cluster (#4785)
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user