ray/release/sgd_tests/sgd_gpu/sgd_gpu_compute.yaml
Antoni Baum 2fb10e6730
[SGD] Add support for native Torch AMP in SGD (#16382)
* SGD native AMP initial commit

* SGD native amp second pass

* Update docs

* Update TorchTrainer doc

* Temp fix release test

* Update release/sgd_tests/sgd_gpu/sgd_gpu_app_config.yaml

Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com>
2021-06-15 17:48:21 -07:00

15 lines
280 B
YAML

cloud_id: {{env["ANYSCALE_CLOUD_ID"]}}
region: us-west-2
max_workers: 1
head_node_type:
name: head_node
instance_type: g3.4xlarge
worker_node_types:
- name: worker_node
instance_type: g3.4xlarge
min_workers: 1
max_workers: 1
use_spot: False