configs.py

from utils import dotdict

configs = {
    'preprocess': False,
    'test-size': 0.2,
    'num-dataset': 2,
    'validation-size': 0.1,
    'train-size': 0.7,
    'test-size': 0.2,
    'batch-size': 32,
    'epochs': 10,
    'learning-rate': 0.001,
    'loss': 'categorical_crossentropy',
    'regularizer': 'l2',
    'regularizer-rate': 0.01,
}

img_configs = {
    'max-size': (1024, 1024),
    'image-size': (128, 128),
    'image-channels': 3,
    'block-size': (32, 32),
    'block-dim': (16, 16),
    'drop-rate': 0.5,
    'image-type': 'float32',
    'image-normalize': True,
    'image-mean': [0.485, 0.456, 0.406],
    'image-std': [0.229, 0.224, 0.225],
    'outlier-rate': 0.0,
}

model_configs = dotdict({
    'dropout': 0.7,
    'num_channels': 256,
    'save_dir': './trainned_models',
    'save_name': 'model'
})