GOT-OCR2_0 / constants.py
ucaslcl's picture
Upload 13 files
8af0d0c verified
raw
history blame
733 Bytes
CONTROLLER_HEART_BEAT_EXPIRATION = 30
WORKER_HEART_BEAT_INTERVAL = 15
LOGDIR = "log"
IGNORE_INDEX = -100
# DEFAULT_PAD_TOKEN = "[PAD]"
DEFAULT_PAD_TOKEN = "<|endoftext|>"
DEFAULT_EOS_TOKEN = "</s>"
DEFAULT_BOS_TOKEN = "</s>"
DEFAULT_UNK_TOKEN = "<unk>"
DEFAULT_IMAGE_TOKEN = "<image>"
DEFAULT_BOX_TOKEN = "<box>"
DEFAULT_IMAGE_PATCH_TOKEN = '<imgpad>'
DEFAULT_IM_START_TOKEN = '<img>'
DEFAULT_IM_END_TOKEN = '</img>'
CONVERSATION_DATA = {
'data_1': {
'images': '/path/',
'annotations': '/path/data1.json',
},
'data_2': {
'images': '/path/',
'annotations': '/path/data2.json',
},
'data_3': {
'images': '/path/',
'annotations': '/path/data3.json',
},
}