mirror of
https://github.com/vale981/ray
synced 2025-03-09 21:06:39 -04:00
56 lines
1.2 KiB
YAML
56 lines
1.2 KiB
YAML
- name: dask_xgboost_test
|
|
owner:
|
|
mail: "antoni@anyscale.com"
|
|
slack: "@team_ml"
|
|
cluster:
|
|
app_config: dask_xgboost_app_config.yaml
|
|
compute_template: compute_tpl.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python workloads/dask_xgboost_test.py
|
|
args:
|
|
[
|
|
"--num-actors 4",
|
|
"--cpus-per-actor 4",
|
|
"--num-actors-inference 16",
|
|
"--cpus-per-actor-inference 1",
|
|
]
|
|
|
|
- name: modin_xgboost_test
|
|
owner:
|
|
mail: "antoni@anyscale.com"
|
|
slack: "@team_ml"
|
|
cluster:
|
|
app_config: modin_xgboost_app_config.yaml
|
|
compute_template: compute_tpl.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python workloads/modin_xgboost_test.py
|
|
args:
|
|
[
|
|
"--num-actors 4",
|
|
"--cpus-per-actor 4",
|
|
"--num-actors-inference 16",
|
|
"--cpus-per-actor-inference 1",
|
|
]
|
|
|
|
- name: torch_tune_serve_test
|
|
owner:
|
|
mail: "matt@anyscale.com"
|
|
slack: "@team_ml"
|
|
|
|
cluster:
|
|
app_config: torch_tune_serve_app_config.yaml
|
|
compute_template: gpu_tpl.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python workloads/torch_tune_serve_test.py
|