for i, key in enumerate(keys): setattr(Settings, key, value_tuple[i]) if not Settings.TEST_ROLLOUT_STATE and Settings.ST_TEST_ROLLOUTS != 2: continue if Settings.ROLLOUT_LENGTH == 1 and Settings.ST_TEST_ROLLOUTS != 2: continue if Settings.ST_TEST_ROLLOUTS > Settings.ROLLOUT_LENGTH: continue do_task() if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument("config", nargs='?', default=None) args = parser.parse_args() config_file = args.config if config_file is not None: Settings.load_from_file(config_file) Settings.setup_logging() merge_gym.register_environments() if Settings.SEED != "Random": np.random.seed(Settings.SEED) torch.manual_seed(Settings.SEED) random.seed(Settings.SEED) if Settings.CUDA: torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False do_task()