# ****HAVE TO INPUT WHETHER CONTINUOUS OR JUST AUTOMATE

all_configs = {
    "asterix": {
        "ANNEAL_LR": True,
        "NUM_ENVS": 64,
        "NUM_STEPS": 128,
        "TOTAL_TIMESTEPS": 1e7,
        "UPDATE_EPOCHS": 4,
        "NUM_MINIBATCHES": 8,
        "GAMMA": 0.99,
        "GAE_LAMBDA": 0.95,
        "CLIP_EPS": 0.2,
        "ENT_COEF": 0.01,
        "VF_COEF": 0.5,
        "MAX_GRAD_NORM": 0.5,
        "ENV_NAME": "Asterix-MinAtar",
        "HSIZE": 64,
        "ACTIVATION": "relu",
        "DEBUG": False,
        "PPO_TEMP": 15.0353,
        "CONTINUOUS": False,
    },
    "freeway": {
        "NUM_ENVS": 64,
        "NUM_STEPS": 128,
        "TOTAL_TIMESTEPS": 1e7,
        "UPDATE_EPOCHS": 4,
        "NUM_MINIBATCHES": 8,
        "GAMMA": 0.99,
        "GAE_LAMBDA": 0.95,
        "CLIP_EPS": 0.2,
        "ENT_COEF": 0.01,
        "VF_COEF": 0.5,
        "MAX_GRAD_NORM": 0.5,
        "ENV_NAME": "Freeway-MinAtar",
        "HSIZE": 64,
        "ACTIVATION": "relu",
        "DEBUG": False,
        "PPO_TEMP": 61.8369,
        "CONTINUOUS": False,
    },
    "breakout": {
        "NUM_ENVS": 64,
        "NUM_STEPS": 128,
        "TOTAL_TIMESTEPS": 5e5,
        "UPDATE_EPOCHS": 4,
        "NUM_MINIBATCHES": 8,
        "GAMMA": 0.99,
        "GAE_LAMBDA": 0.95,
        "CLIP_EPS": 0.2,
        "ENT_COEF": 0.01,
        "VF_COEF": 0.5,
        "MAX_GRAD_NORM": 0.5,
        "ENV_NAME": "Breakout-MinAtar",
        "HSIZE": 64,
        "ACTIVATION": "relu",
        "DEBUG": False,
        "PPO_TEMP": 50.0822,
        "CONTINUOUS": False,
    },
    "spaceinvaders": {
        "NUM_ENVS": 64,
        "NUM_STEPS": 128,
        "TOTAL_TIMESTEPS": 1e7,
        "UPDATE_EPOCHS": 4,
        "NUM_MINIBATCHES": 8,
        "GAMMA": 0.99,
        "GAE_LAMBDA": 0.95,
        "CLIP_EPS": 0.2,
        "ENT_COEF": 0.01,
        "VF_COEF": 0.5,
        "MAX_GRAD_NORM": 0.5,
        "ENV_NAME": "SpaceInvaders-MinAtar",
        "HSIZE": 64,
        "ACTIVATION": "relu",
        "DEBUG": False,
        "PPO_TEMP": 167.0913,
        "CONTINUOUS": False,
    },
    "ant": {
        "NUM_ENVS": 2048,
        "NUM_STEPS": 10,
        "TOTAL_TIMESTEPS": 5e7,
        "UPDATE_EPOCHS": 4,
        "NUM_MINIBATCHES": 32,
        "GAMMA": 0.99,
        "GAE_LAMBDA": 0.95,
        "CLIP_EPS": 0.2,
        "ENT_COEF": 0.0,
        "VF_COEF": 0.5,
        "MAX_GRAD_NORM": 0.5,
        "ACTIVATION": "tanh",
        "HSIZE": 64,
        "ENV_NAME": "Brax-ant",
        "BACKEND": "positional",
        "SYMLOG_OBS": False,
        "CLIP_ACTION": True,
        "DEBUG": False,
        "NORMALIZE": True,
        "CONTINUOUS": True,
        "PPO_TEMP": 6517.31,
    },
    "gridworld": {
        "LR": 1e-4,
        "ANNEAL_LR": True,
        "NUM_ENVS": 1024,
        "NUM_STEPS": 20,
        "TOTAL_TIMESTEPS": 3e7,
        "UPDATE_EPOCHS": 2,
        "NUM_MINIBATCHES": 16,
        "GAMMA": 0.99,
        "GAE_LAMBDA": 0.95,
        "CLIP_EPS": 0.2,
        "ENT_COEF": 0.01,
        "VF_COEF": 0.5,
        "MAX_GRAD_NORM": 0.5,
        "ACTIVATION": "tanh",
        "HSIZE": 16,
        "ENV_NAME": "gridworld",
        "DEBUG": False,
        "PPO_TEMP": 1.0,
        "CONTINUOUS": False,
    },
}
