ImitationRunner

class maze.train.trainers.imitation.imitation_runners.ImitationRunner(state_dict_dump_file: str, spaces_config_dump_file: str, normalization_samples: int, dataset: omegaconf.DictConfig)

Dev runner for imitation learning.

Loads the given trajectory data and trains a policy on top of it using supervised learning.

dataset: omegaconf.DictConfig

Specify the Dataset class used to load the trajectory data for training

run(cfg: omegaconf.DictConfig) → None

Run the training master node.