mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
e1e4a45b8d
* [train] add simple Ray Train release tests * simplify tests * update * driver requirements * move to test * remove connect * fix * fix * fix torch * gpu * add assert * remove assert * use gloo backend * fix * finish Co-authored-by: Amog Kamsetty <amogkamsetty@yahoo.com>
17 lines
447 B
YAML
17 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
|