Skip to content

Commit

Permalink
Ray readme fix (#96)
Browse files Browse the repository at this point in the history
  • Loading branch information
al-rigazzi committed Oct 13, 2021
1 parent ebd72b0 commit f1d28dc
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions README.md
Expand Up @@ -97,7 +97,7 @@ independently.
- [Online Processing](#online-processing)
- [Singular Value Decomposition](#singular-value-decomposition)
- [Online Inference](#online-inference)
- [PyTorch CNN Example](#pytorch)
- [PyTorch CNN Example](#pytorch-cnn-example)
- [Publications](#publications)
- [Cite](#cite)
- [bibtex](#bibtex)
Expand Down Expand Up @@ -445,19 +445,19 @@ allocated nodes.
import ray

from smartsim import Experiment
from smartsim.ext.ray import RayCluster
from smartsim.exp.ray import RayCluster

exp = Experiment("ray-cluster", launcher='slurm')
# 3 workers + 1 head node = 4 node-cluster
cluster = RayCluster(name="ray-cluster", run_args={},
ray_args={"num-cpus": 24},
launcher=launcher, workers=3, batch=True)
launcher='slurm', num_nodes=4, batch=True)

exp.generate(cluster, overwrite=True)
exp.start(cluster, block=False, summary=True)

# Connect to the Ray cluster
ray.util.connect(cluster.head_model.address+":10001")
ctx = ray.init("ray://"+cluster.get_head_address()+":10001")

# <run Ray tune, RLlib, HPO...>
```
Expand All @@ -475,19 +475,19 @@ allocated nodes.
import ray

from smartsim import Experiment
from smartsim.ext.ray import RayCluster
from smartsim.exp.ray import RayCluster

exp = Experiment("ray-cluster", launcher='pbs')
# 3 workers + 1 head node = 4 node-cluster
cluster = RayCluster(name="ray-cluster", run_args={},
ray_args={"num-cpus": 24},
launcher=launcher, workers=3, batch=True)
launcher='pbs', num_nodes=4, batch=True)

exp.generate(cluster, overwrite=True)
exp.start(cluster, block=False, summary=True)

# Connect to the ray cluster
ray.util.connect(cluster.head_model.address+":10001")
ctx = ray.init("ray://"+cluster.get_head_address()+":10001")

# <run Ray tune, RLlib, HPO...>
```
Expand Down

0 comments on commit f1d28dc

Please sign in to comment.