mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
14 lines
387 B
YAML
14 lines
387 B
YAML
|
# Test multi-node, multi-GPU Ray SGD example.
|
||
|
|
- name: sgd_gpu
|
||
|
|
owner:
|
||
|
|
mail: "amog@anyscale.com"
|
||
|
|
slack: "@tune-team"
|
||
|
|
|
||
|
|
cluster:
|
||
|
|
app-config: sgd_gpu/sgd_gpu_app_config.yaml
|
||
|
|
compute_template: sgd_gpu/sgd_gpu_compute.yaml
|
||
|
|
|
||
|
|
run:
|
||
|
|
timeout: 3000
|
||
|
|
prepare: python wait_cluster.py 2 600
|
||
|
|
script: python sgd_gpu/sgd_gpu_test.py --num-workers=2 --use-gpu --address=auto
|