hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[AIR] Fix trainer allowed scaling config keys (#25350)

Browse source 
Adds `resources_per_worker` to allowed scaling config keys in `DataParallelTrainer` and `GBDTTrainer`.
This commit is contained in:
Antoni Baum 2022-06-02 20:20:37 +02:00 • committed by GitHub
parent 16bdfe6a39
commit f8551942bf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
python/ray/ml/train/data_parallel_trainer.py
View file

@ -228,6 +228,7 @@ class DataParallelTrainer(Trainer):
"num_workers",
"num_cpus_per_worker",
"num_gpus_per_worker",
"resources_per_worker",
"additional_resources_per_worker",
"use_gpu",
]

1
python/ray/ml/train/gbdt_trainer.py
View file

@ -69,6 +69,7 @@ class GBDTTrainer(Trainer):
"num_workers",
"num_cpus_per_worker",
"num_gpus_per_worker",
"resources_per_worker",
"additional_resources_per_worker",
"use_gpu",
]