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)