Config.loss_gamma = 1. # dataset Config.num_classes = 20 Config.batch_size = 16 # 128 Config.root = '../../dataset/voc/VOCdevkit/VOC2012' Config.split = 'trainval' Config.resize_size = [512, 512] Config.num_workers = 4 Config.mean = [0.40789654, 0.44719302, 0.47026115] Config.std = [0.28863828, 0.27408164, 0.27809835] # train Config.optimizer = 'AdamW' Config.lr = 1e-3 Config.AMSGRAD = True Config.max_iter = 45000 Config.lr_schedule = 'WarmupMultiStepLR' Config.gamma = 0.1 Config.steps = (35000, 40000) Config.warmup_iters = 1000 Config.apex = False # other Config.gpu = True Config.eval = False Config.resume = False Config.score_th = 0.1 Config.down_stride = 4
