class Config:
    def __init__(self,
                 batch_size=32,
                 #device="cuda:1",
                 device="cpu",
                 learning_rate=0.01,
                 learning_rate_decay=1,
                 epsilon = 0.1,
                 epsilon_low = 0.1,
                 epsilon_step = 0.05,
                 epochs=1000,
                 random_reset = True,
                 max_play_length=200,
                 gamma = 0.99
                 ):

        self.batch_size = batch_size
        self.device=device
        self.learning_rate=learning_rate
        self.learning_rate_decay=learning_rate_decay
        self.epsilon = epsilon
        self.epsilon_low = epsilon_low
        self.epsilon_step = epsilon_step
        self.epochs=epochs
        self.random_reset = random_reset
        self.max_play_length=max_play_length
        self.gamma = gamma
