search

google-deepmind / launchpad

Apache License 2.0
309 stars 35 forks source link

Example of how to specify local resources. #4

Closed arnupretorius closed 3 years ago

arnupretorius commented 3 years ago

Hi there, thanks for releasing this! Really appreciated :)

I was just wondering, is there perhaps an example for how to specify the local resources for a launch? I can see that launch allows for an optional Dict[str, Any] to set up the resource configs. I'm guessing here the str keys refer to the nodes in the program but I'm not sure how to specify the "per-node launch configuration" for each node. Would this also be a dictionary, e.g. {"gpus": 2, "cpus": 4}? Apologies if there are examples of this that I missed.

qstanczyk commented 3 years ago

Hi Arnu, local_resources accepts a dictionary of {"node group" -> PythonProcess }