mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
18 lines
447 B
YAML
18 lines
447 B
YAML
|
- name: train_tensorflow_mnist_test
|
||
|
|
cluster:
|
||
|
|
app_config: app_config.yaml
|
||
|
|
compute_template: compute_tpl.yaml
|
||
|
|
|
||
|
|
run:
|
||
|
|
timeout: 36000
|
||
|
|
script: python workloads/train_tensorflow_mnist_test.py
|
||
|
|
|
||
|
|
- name: train_torch_linear_test
|
||
|
|
cluster:
|
||
|
|
app_config: app_config.yaml
|
||
|
|
compute_template: compute_tpl.yaml
|
||
|
|
|
||
|
|
run:
|
||
|
|
timeout: 36000
|
||
|
|
script: python workloads/train_torch_linear_test.py
|