ksuit.runners.hydra_runner

Attributes

logger

Classes

HydraRunner

Runs an experiment using @hydra.main as entry point.

Module Contents

ksuit.runners.hydra_runner.logger
class ksuit.runners.hydra_runner.HydraRunner

Runs an experiment using @hydra.main as entry point.

run(config)

Runs an experiment. Given CLI args which contain a path to a hyperparameter file.

Parameters:

config (dict) – hydra config as a YAML.

Return type:

None

static derive_run_name(name)

Derives a run name by appending hydra overrides to the given name.

Parameters:

name (str)

static main(device, config, static_config)

Main method called from each GPU main process after being spawned and initialized for communication.

Parameters:

  • device (torch.device) – device of the current process.

  • config (ksuit.schemas.schema.ConfigSchema) – configuration of the experiment.

  • static_config (ksuit.configs.StaticConfig)

Return type:

None