torchrl.experiments package

Submodules

torchrl.experiments.base_experiment module

class torchrl.experiments.base_experiment.BaseExperiment(env_id: str = None, n_envs: int = 1, n_frames: int = 1000, n_rand_frames: int = 0, n_train_interval: int = 100, **kwargs)[source]

Bases: kondo.experiment.Experiment

act(obs_list: list) → list[source]
build_controller() → torchrl.controllers.controller.Controller[source]
run()[source]
store(transition_list)[source]

Placeholder method for storage related usage.

train() → dict[source]

Placeholder method for training related usage.

torchrl.experiments.base_experiment.log_dict(logger, info: dict, tag: str, step=None)[source]

torchrl.experiments.test_base_experiment module

torchrl.experiments.test_base_experiment.test_base_exp(spec_id: str, n_frames: str)[source]

Module contents