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()
