CONF.TRAIN.VAL_SIZE = -1 CONF.TRAIN.LEARNING_RATE = 2e-4 CONF.TRAIN.WEIGHT_DECAY = 5e-4 CONF.TRAIN.VERBOSE = 10 # fixed batch size if CONF.TRAIN.DATASET == 'shapenet': CONF.TRAIN.BATCH_SIZE = 100 elif CONF.TRAIN.DATASET == 'primitives': CONF.TRAIN.BATCH_SIZE = 75 else: raise ValueError("invalid dataset, terminating...") ''' global configurations for BERT ''' CONF.BERT = EasyDict() # flags of BERT CONF.BERT.IS_BERT = True # configs of BERT CONF.BERT.MODEL = "bert-large-uncased" CONF.BERT.DIM = 1024 ''' global configurations for LBA ''' CONF.LBA = EasyDict() # flags of LBA CONF.LBA.IS_LBA = True CONF.LBA.IS_LBA_TST = True CONF.LBA.IS_LBA_STS = True CONF.LBA.IS_LBA_VISIT = True