loss_type:
t5seq_pretrain_margin_mse
t5seq_aq_encoder_seq2seq
t5seq_aq_encoder_margin_mse
t5seq_aq_encoder_margin_mse_sub_smtid
t5seq_aq_encoder_lng_knp_margin_mse

evaluate:
mmap
aq_index
aq_evaluate
aq_to_flat_index_search_evaluate
retrieve_train_queries
index 
retrieve
t5seq_aq_get_qid_to_smtid_rankdata
t5seq_aq_retrieve_docids_use_sub_smtid

model:
T5SeqPretrainEncoder