constants.py

# Model constants
PSPNET = 'pspnet'
PSPNET_50 = 'pspnet_50'
VGG_19 = 'vgg_19'
VGG_16 = 'vgg'
RESNET = 'resnet'
RESNET_50 = 'resnet_50'

# Loss functions
CE_LOSS = 'ce_loss'

# Dataset
TEMPLES_DATASET = 'temples_dataset'

# Optimizers
SGD = 'sgd'
AMSGRAD = 'amsgrad'
ADAM = 'adam'
RMSPROP = 'rmsprop'
ADABOUND = 'adabound'

# Normalization layers
INSTANCE_NORM = 'instance'
BATCH_NORM = 'batch'
SYNC_BATCH_NORM = 'syncbn'

# Init types
NORMAL_INIT = 'normal'
KAIMING_INIT = 'kaiming'
XAVIER_INIT = 'xavier'
ORTHOGONAL_INIT = 'orthogonal'

# Downsampling methods
MAXPOOL = 'maxpool'
STRIDECONV = 'strideconv'

# Split constants
TRAIN = 'train'
VAL = 'val'
TEST = 'test'
TRAINVAL = 'trainval'
VISUALIZATION = 'visualization'