import torch from attr_dict import AttrDict __C = AttrDict() cfg = __C __C.GLOBAL_RANK = 0 __C.EPOCH = 0 # Absolute path to a location to keep some large files, not in this dir. __C.ASSETS_PATH = '/home/dcg-adlr-atao-data.cosmos277/assets' # Use class weighted loss per batch to increase loss for low pixel count classes per batch __C.BATCH_WEIGHTING = False # Border Relaxation Count __C.BORDER_WINDOW = 1 # Number of epoch to use before turn off border restriction __C.REDUCE_BORDER_EPOCH = -1 # Comma Seperated List of class id to relax __C.STRICTBORDERCLASS = None # Where output results get written __C.RESULT_DIR = None __C.OPTIONS = AttrDict() __C.OPTIONS.TEST_MODE = False __C.OPTIONS.INIT_DECODER = False __C.OPTIONS.TORCH_VERSION = None __C.TRAIN = AttrDict() __C.TRAIN.RANDOM_BRIGHTNESS_SHIFT_VALUE = 10 __C.TRAIN.FP16 = False
