torchrl.cli.commands package

Submodules

torchrl.cli.commands.list module

torchrl.cli.commands.resume module

torchrl.cli.commands.run module

class torchrl.cli.commands.run.ExtraHParamType[source]

Bases: click.types.ParamType

convert(value, param, ctx)[source]

Converts the value. This is not invoked for values that are None (the missing value).

torchrl.cli.commands.run.do_run(problem, hparam_set: str = None, extra_hparams: dict = None, progress: bool = False, cuda: bool = False, device: str = None, log_dir: str = None, resume: bool = False, start_epoch: int = None, checkpoint_prefix: str = 'checkpoint', **kwargs)[source]

Module contents