scripts/training/config.py

BLOCK_OPTIONS = [8, 16, 32, 64, 128]

class NetworkConfig:
    def __init__(self):
        self.block1 = 32
        self.block2 = 64
        self.block3 = 128
        self.batch_size = 64
        self.optimizer = 'SGD'
        self.epochs = 1

    def update(self, **kwargs):
        for key, value in kwargs.items():
            if hasattr(self, key):
                setattr(self, key, value)