cfg = __C __C.AUTO_RESUME = True __C.DATA_DIR = './data/imagenet' __C.MODEL = 'cream' __C.RESUME_PATH = './experiments/ckps/resume.pth.tar' __C.SAVE_PATH = './experiments/ckps/' __C.SEED = 42 __C.LOG_INTERVAL = 50 __C.RECOVERY_INTERVAL = 0 __C.WORKERS = 4 __C.NUM_GPU = 1 __C.SAVE_IMAGES = False __C.AMP = False __C.ACC_GAP = 5 __C.OUTPUT = 'output/path/' __C.EVAL_METRICS = 'prec1' __C.TTA = 0 # Test or inference time augmentation __C.LOCAL_RANK = 0 __C.VERBOSE = False # dataset configs __C.DATASET = CN() __C.DATASET.NUM_CLASSES = 1000 __C.DATASET.IMAGE_SIZE = 224 # image patch size __C.DATASET.INTERPOLATION = 'bilinear' # Image resize interpolation type __C.DATASET.BATCH_SIZE = 32 # batch size __C.DATASET.NO_PREFECHTER = False __C.DATASET.PIN_MEM = True __C.DATASET.VAL_BATCH_MUL = 4
