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
