config.py

from dataclasses import dataclass


@dataclass
class Config:
    gamma: float
    batch_size: int
    lr: float
    initial_exploration: int
    log_interval: int
    update_target: int
    replay_memory_capacity: int
    device: str
    sequence_length: int
    burn_in_length: int
    eta: float
    local_mini_batch: int
    n_step: int
    over_lapping_length: int
    epsilon_decay: float
    random_seed: int
    enable_ngu: bool
    hidden_size: int
    L: float
    beta: float
    n_actors: int
    eps: float
    alpha: int



config = Config(
    gamma=0.99,
    batch_size=32,
    lr=0.001,
    initial_exploration=1000,
    log_interval=10,
    update_target=10,
    replay_memory_capacity=1000000,
    device="cpu",
    sequence_length=32,
    burn_in_length=4,
    eta=0.9,
    local_mini_batch=8,
    n_step=2,
    over_lapping_length=16,
    epsilon_decay=0.00001,
    random_seed=42,
    enable_ngu=True,
    hidden_size=16,
    L=5.0,
    beta=0.3,
    n_actors=8,
    eps=0.4,
    alpha=8
)