narla.settings

Settings

class narla.settings.Settings(environment_settings=<factory>, multi_agent_network_settings=<factory>, trial_settings=<factory>)[source]

Bases: narla.settings.base_settings.BaseSettings

environment_settings: narla.environments.environment_settings.EnvironmentSettings

multi_agent_network_settings: narla.multi_agent_network.multi_agent_network_settings.MultiAgentNetworkSettings

trial_settings: narla.settings.trial_settings.TrialSettings

TrialSettings

class narla.settings.TrialSettings(batch_size=128, device='cpu', gpu=0, maximum_episodes=20000, random_seed=0, results_directory='', save_every=1000, trial_id=0)[source]

Bases: narla.settings.base_settings.BaseSettings

batch_size: int = 128: Batch size to use during training

device: Literal['cpu', 'cuda'] = 'cpu': Device to put the network on

gpu: int = 0: GPU ID to run on

maximum_episodes: int = 20000: Total number of episodes to run for

random_seed: int = 0: Random seed

results_directory: str = '': Path to save results

save_every: int = 1000: Save results every n steps

trial_id: int = 0: Unique ID of the trial being run, corresponds to the path of data saving <results_directory>/<trial_id>/