Struct FRLlibResourceSettings

struct FRLlibResourceSettings : public FTrainingSettings

A struct to hold resource settings for an RLlib training script.

Public Functions

virtual void GenerateTrainingArgs(int Port, FScriptArgBuilder &ArgBuilder) const

Generate the training arguments for the script using the ArgBuilder.

Note

port is supplied as it is a common argument to pass to scripts, and is set at a high level but might be needed by specific subsettings

Parameters:

• Port – [in] The port to use for the script

• ArgBuilder – [in] The builder to use to generate the arguments

virtual ~FRLlibResourceSettings()

Public Members

int NumGPUs = 0

The number of GPUs to use for training.

bool bUseCustomNumberOfCPUs = false

Whether to use a custom number of CPUs for training.

int NumCPUs = 1

The maximum number of CPUs/workers to use for training.

If empty uses number of vCPUs on system

int NumLearners = 0

The number of learner processes to use for training.

Set to 0 to run in main process

int NumGPUsPerLearner = 0

The number of GPUs to use for the learner.

int NumCPUsPerLearner = 1

The number of CPUs to use for the learner.

int NumCPUsForMainProcess = 1

The number of CPUs to use for the main process.