def main(cfg): print(cfg.pretty(), end="\n\n\n") environment = get_environment(**cfg["environment"]) agent = Agent.from_conf(environment, **cfg["agent"]) replay_buffer = ReplayBufferBase.from_conf(**cfg["replay_buffer"]) algorithm = Algorithm.from_conf(environment, agent, replay_buffer, **cfg["algorithm"]) algorithm()
def main(cfg): print(cfg.pretty(), end="\n\n\n") config_path = cfg.experiment_path + "/.hydra/config.yaml" exp_cfg = omegaconf.OmegaConf.load(config_path) exp_cfg.environment.monitor = "always" environment = get_environment(**exp_cfg["environment"]) agent = Agent.from_conf(environment, **exp_cfg["agent"]) replay_buffer = ReplayBuffer.from_conf(**exp_cfg["replay_buffer"]) algorithm = Algorithm.from_conf(environment, agent, replay_buffer, **exp_cfg["algorithm"]) algorithm.restore_model(cfg.experiment_path + "/checkpoints") for i in range(cfg["n_episodes"]): print("{: 3d}/{: 3d}".format(i + 1, cfg["n_episodes"]), end='\r') algorithm.evaluate()