def get_config():
    config = {
        "episodes": 0,
        "eps_start": 0.9,
        "eps_end": 0.05,
        "eps_decay": 200,
        "gamma": 0.8,
        "dqn_lr": 1e-3,
        "policy_hidden_size": 256,
        "q_network_batch_size": 64,
        "using_symmetry": True,
        "learned_symmetry": True,
        "env": "CartPole-v0",
        "lens_hidden_size": 256,
        "lens_learning_rate": 1e-3,
        "transition_model_hidden_size": 16,
        "replay_size": 10000,
        "lens_batch_size": 16,
        "lens_training_steps": 20000,
        "data_collection_episodes": 3,
        "canonical_action": 0,
        "lens_action": 1,
    }
    return config
