Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Mandar Joshi
committed
Jun 14, 2019
1 parent
8a2e8c1
commit ae74353
Showing
1 changed file
with
39 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,39 @@ | ||
CURRENT_MODELS = { | ||
'google_base_cased': ('bert-base-cased', '/checkpoint/mandarj/coref_data/bert_models/cased_L-12_H-768_A-12/bert_model.ckpt'), | ||
'base_cased_random': ('bert-base-cased', '/private/home/omerlevy/cookie/cased_models//base_random/checkpoint_23_1000000.pt'), | ||
'base_cased_word': ('bert-base-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/base_word/checkpoint_23_1000000.pt'), | ||
'base_cased_geo_span': ('bert-base-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/base_geo_span/checkpoint_23_1000000.pt'), | ||
'google_base_uncased': ('bert-base-uncased', '/checkpoint/mandarj/coref_data/bert_models/uncased_L-12_H-768_A-12/bert_model.ckpt'), | ||
'base_uncased_random': ('bert-base-uncased', '/private/home/mandarj/workspace/huggingface-fairseq/final/uncased_base_random/checkpoint_24_1000000.pt'), | ||
'base_uncased_word': ('bert-base-uncased', '/private/home/omerlevy/cookie/uncased_models/base_word/checkpoint_24_1000000.pt'), | ||
'base_uncased_geo_span': ('bert-base-uncased', '/private/home/omerlevy/cookie/uncased_models/base_geo_span/checkpoint_24_1000000.pt'), | ||
'google_large_cased': ('bert-large-cased', '/checkpoint/mandarj/coref_data/bert_models/cased_L-24_H-1024_A-16/bert_model.ckpt'), | ||
'large_cased_random': ('bert-large-cased', '/private/home/omerlevy/cookie/cased_models/random/checkpoint_23_1000000.pt'), | ||
'large_cased_word': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/word/checkpoint_23_1000000.pt'), | ||
'large_cased_geo_span': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/geo_span/checkpoint_23_1000000.pt'), | ||
'large_cased_geo_span_0.2': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/geo_span_0.2/checkpoint_23_1000000.pt'), | ||
'large_cased_ner_span': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/ner_span/checkpoint_23_1000000.pt'), | ||
'large_cased_np_span': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/np_span/checkpoint_23_1000000.pt'), | ||
'large_cased_no_nsp_random': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/no_nsp_random/checkpoint_31_1000000.pt'), | ||
'large_cased_no_nsp_geo_span': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/no_nsp_geo_span/checkpoint_31_1000000.pt'), | ||
'large_cased_no_nsp_geo_span_0.2': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/no_nsp_0.2/checkpoint_31_1000000.pt'), | ||
'large_cased_no_nsp_pair': ('bert-large-cased', '/private/home/mandarj/workspace/huggingface-fairseq/final/no_nsp_pair/checkpoint_31_1000000.pt'), | ||
'google_large_uncased': ('bert-large-uncased', '/checkpoint/mandarj/coref_data/bert_models/uncased_L-24_H-1024_A-16/bert_model.ckpt'), | ||
'large_uncased_random': ('bert-large-uncased', '/private/home/mandarj/workspace/huggingface-fairseq/final/uncased_random/checkpoint_24_1000000.pt'), | ||
'large_uncased_word': ('bert-large-uncased', '/private/home/omerlevy/cookie/uncased_models/word/checkpoint_24_1000000.pt'), | ||
'large_uncased_geo_span': ('bert-large-uncased', '/private/home/omerlevy/cookie/uncased_models/geo_span/checkpoint_24_1000000.pt'), | ||
'large_uncased_ner_span': ('bert-large-uncased', '/private/home/omerlevy/cookie/uncased_models/ner_span/checkpoint_24_1000000.pt'), | ||
'large_uncased_np_span': ('bert-large-uncased', '/private/home/omerlevy/cookie/uncased_models/np_span/checkpoint_24_1000000.pt'), | ||
'large_uncased_no_nsp_random': ('bert-large-uncased', '/private/home/omerlevy/cookie/uncased_models/no_nsp_random/checkpoint_32_1000000.pt'), | ||
'large_uncased_no_nsp_geo_span': ('bert-large-uncased', '/private/home/omerlevy/cookie/uncased_models/no_nsp_geo_span/checkpoint_32_1000000.pt'), | ||
'large_uncased_no_nsp_geo_span_0.2': ('bert-large-uncased', '/private/home/mandarj/workspace/huggingface-fairseq/final/uncased_no_nsp_0.2/checkpoint_32_1000000.pt'), | ||
'marjan_large_uncased': ('bert-large-uncased', '/checkpoint/ghazvini/geo_large//geol4/checkpoint_24_960000.pt'), | ||
'large_uncased_geo_span_no_clamp': ('bert-large-uncased', '/private/home/mandarj/workspace/huggingface-fairseq/data/lspan_replace/checkpoint_24_1000000.pt'), | ||
'yinhan_base_uncased': ('bert-base-uncased', '/checkpoint/yinhanliu/20190329/hf_bert_implement.4096token/bert512.eps-08_0.00025/checkpoint_23_960000.pt'), | ||
'yinhan_large_uncased': ('bert-large-uncased', '/checkpoint/yinhanliu/bert_large/checkpoint_23_960000.pt'), | ||
'danqi_base_uncased_1024': ('bert-base-uncased-1024', '/checkpoint/danqi/20190523/mlm_doc_1024_0.00025/checkpoint32.pt'), | ||
'danqi_base_uncased_512': ('bert-base-uncased', '/checkpoint/danqi/20190530/mlm_doc_512_0.00025/checkpoint32.pt'), | ||
'danqi_base_uncased': ('bert-base-uncased', '/checkpoint/danqi/20190530/mlm_doc_512_0.00025/checkpoint_62_1950000_fixed.pt') | ||
} | ||
|
||
MODEL_CAT_TO_GOOGLE_DIR = {'bert-base-cased': 'cased_L-12_H-768_A-12', 'bert-base-uncased': 'uncased_L-12_H-768_A-12', 'bert-large-cased': 'cased_L-24_H-1024_A-16', 'bert-large-uncased': 'uncased_L-24_H-1024_A-16', 'bert-base-uncased-1024': 'uncased_L-12_H-768_A-12'} |