from easydict import EasyDict cfg = EasyDict() cfg.CLS_FC = [256, 128] cfg.AFF_FC = [512, 256, 256, 128] cfg.SHARED_FC1 = [512] cfg.SHARED_FC2 = [256, 256, 128] cfg.LOCAL_RANK = 0 cfg.OPTIMIZATION = EasyDict() cfg.OPTIMIZATION.BATCH_SIZE_PER_GPU = 15 cfg.OPTIMIZATION.NUM_EPOCHS = 100 cfg.OPTIMIZATION.OPTIMIZER = 'adam' cfg.OPTIMIZATION.LR = 0.0005 cfg.OPTIMIZATION.WEIGHT_DECAY = 0.001 cfg.OPTIMIZATION.MOMENTUM = 0.9 cfg.OPTIMIZATION.MOMS = [0.95, 0.85] cfg.OPTIMIZATION.PCT_START = 0.4 cfg.OPTIMIZATION.DIV_FACTOR = 10 cfg.OPTIMIZATION.DECAY_STEP_LIST = [35, 45] cfg.OPTIMIZATION.LR_DECAY = 0.1 cfg.OPTIMIZATION.LR_CLIP = 0.0000001 cfg.OPTIMIZATION.LR_WARMUP = False cfg.OPTIMIZATION.WARMUP_EPOCH = 1