Use custom resources for PS machines.

diux-dev / cluster

train on AWS

75 stars 15 forks source link

Use custom resources for PS machines. #7

Closed robertnishihara closed 6 years ago

robertnishihara commented 6 years ago

Also need to modify the launching script so that we pass --resources='{"PSMachine": 1}' for each of the parameter server machines (and get rid of --num-gpus).