from utils import ShapeSpec cfg = ED() # TODO: Most hyperparameters be hard code(so I can test it easy), it all will be added to this file later(maybe). # trainer # epoch, batch_size, so on. # base info cfg.NUM_CLASSES = 1 cfg.EPOCH = 100 cfg.lr = 1e-4 # data cfg.ROOT = 'datasets/wgisd' cfg.BATCH_SIZE = 3 cfg.NUM_WORKERS = 8 cfg.RESIZE = (800, 1280) # backbone cfg.BACKBONE_DEPTH = 101 cfg.BACKBONE_OUTPUT_SHAPE = { 'res2': ShapeSpec(channels=256, height=None, width=None, stride=4), 'res3': ShapeSpec(channels=512, height=None, width=None, stride=8), 'res4': ShapeSpec(channels=1024, height=None, width=None, stride=16), 'res5': ShapeSpec(channels=2048, height=None, width=None, stride=32) } # MASK cfg.CLS_AGNOSTIC_MASK = False