Upload folder using huggingface_hub
Browse files- artifact.metadata +73 -0
 - config.json +25 -0
 - model.safetensors +3 -0
 - special_tokens_map.json +37 -0
 - tokenizer_config.json +63 -0
 - vocab.txt +0 -0
 
    	
        artifact.metadata
    ADDED
    
    | 
         @@ -0,0 +1,73 @@ 
     | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 1 | 
         
            +
            {
         
     | 
| 2 | 
         
            +
                "query_token_id": "[unused0]",
         
     | 
| 3 | 
         
            +
                "doc_token_id": "[unused1]",
         
     | 
| 4 | 
         
            +
                "query_token": "[\u8cea\u554fs]",
         
     | 
| 5 | 
         
            +
                "doc_token": "[\u6587\u66f8]",
         
     | 
| 6 | 
         
            +
                "ncells": null,
         
     | 
| 7 | 
         
            +
                "centroid_score_threshold": null,
         
     | 
| 8 | 
         
            +
                "ndocs": null,
         
     | 
| 9 | 
         
            +
                "load_index_with_mmap": false,
         
     | 
| 10 | 
         
            +
                "index_path": null,
         
     | 
| 11 | 
         
            +
                "index_bsize": 64,
         
     | 
| 12 | 
         
            +
                "nbits": 2,
         
     | 
| 13 | 
         
            +
                "kmeans_niters": 4,
         
     | 
| 14 | 
         
            +
                "resume": false,
         
     | 
| 15 | 
         
            +
                "pool_factor": 1,
         
     | 
| 16 | 
         
            +
                "clustering_mode": "hierarchical",
         
     | 
| 17 | 
         
            +
                "protected_tokens": 0,
         
     | 
| 18 | 
         
            +
                "similarity": "cosine",
         
     | 
| 19 | 
         
            +
                "bsize": 16,
         
     | 
| 20 | 
         
            +
                "accumsteps": 1,
         
     | 
| 21 | 
         
            +
                "lr": 3e-5,
         
     | 
| 22 | 
         
            +
                "maxsteps": 50000,
         
     | 
| 23 | 
         
            +
                "save_every": null,
         
     | 
| 24 | 
         
            +
                "warmup": 2500,
         
     | 
| 25 | 
         
            +
                "warmup_bert": null,
         
     | 
| 26 | 
         
            +
                "relu": false,
         
     | 
| 27 | 
         
            +
                "nway": 32,
         
     | 
| 28 | 
         
            +
                "use_ib_negatives": false,
         
     | 
| 29 | 
         
            +
                "reranker": false,
         
     | 
| 30 | 
         
            +
                "distillation_alpha": 1.0,
         
     | 
| 31 | 
         
            +
                "ignore_scores": false,
         
     | 
| 32 | 
         
            +
                "model_name": "ja_colbert_v2.4",
         
     | 
| 33 | 
         
            +
                "schedule_free": true,
         
     | 
| 34 | 
         
            +
                "schedule_free_wd": 0.0,
         
     | 
| 35 | 
         
            +
                "kldiv_loss": true,
         
     | 
| 36 | 
         
            +
                "marginmse_loss": false,
         
     | 
| 37 | 
         
            +
                "kldiv_weight": 1.0,
         
     | 
| 38 | 
         
            +
                "marginmse_weight": 0.05,
         
     | 
| 39 | 
         
            +
                "ib_loss_weight": 1.0,
         
     | 
| 40 | 
         
            +
                "normalise_training_scores": true,
         
     | 
| 41 | 
         
            +
                "normalization_method": "minmax",
         
     | 
| 42 | 
         
            +
                "quant_aware": false,
         
     | 
| 43 | 
         
            +
                "highest_quant_level": 8,
         
     | 
| 44 | 
         
            +
                "lowest_quant_level": 2,
         
     | 
| 45 | 
         
            +
                "query_maxlen": 32,
         
     | 
| 46 | 
         
            +
                "attend_to_mask_tokens": false,
         
     | 
| 47 | 
         
            +
                "interaction": "colbert",
         
     | 
| 48 | 
         
            +
                "cap_padding": 0,
         
     | 
| 49 | 
         
            +
                "dim": 128,
         
     | 
| 50 | 
         
            +
                "doc_maxlen": 300,
         
     | 
| 51 | 
         
            +
                "mask_punctuation": true,
         
     | 
| 52 | 
         
            +
                "checkpoint": "bclavie\/JaColBERT",
         
     | 
| 53 | 
         
            +
                "triples": "\/home\/azureuser\/jp\/data\/triplets\/full\/normalized\/m3.jsonl",
         
     | 
| 54 | 
         
            +
                "collection": "\/home\/azureuser\/jp\/data\/mmarco\/collection.tsv",
         
     | 
| 55 | 
         
            +
                "queries": "\/home\/azureuser\/jp\/data\/mmarco\/queries.tsv",
         
     | 
| 56 | 
         
            +
                "index_name": null,
         
     | 
| 57 | 
         
            +
                "overwrite": false,
         
     | 
| 58 | 
         
            +
                "root": "\/home\/azureuser\/jp\/experiments",
         
     | 
| 59 | 
         
            +
                "experiment": "jacolbert_v2.5",
         
     | 
| 60 | 
         
            +
                "index_root": null,
         
     | 
| 61 | 
         
            +
                "name": "JaColBERTv2.5",
         
     | 
| 62 | 
         
            +
                "rank": 0,
         
     | 
| 63 | 
         
            +
                "nranks": 4,
         
     | 
| 64 | 
         
            +
                "amp": true,
         
     | 
| 65 | 
         
            +
                "gpus": 4,
         
     | 
| 66 | 
         
            +
                "avoid_fork_if_possible": false,
         
     | 
| 67 | 
         
            +
                "meta": {
         
     | 
| 68 | 
         
            +
                    "hostname": "a100-80-4x",
         
     | 
| 69 | 
         
            +
                    "current_datetime": "",
         
     | 
| 70 | 
         
            +
                    "cmd": "",
         
     | 
| 71 | 
         
            +
                    "version": "colbert-v0.4"
         
     | 
| 72 | 
         
            +
                }
         
     | 
| 73 | 
         
            +
            }
         
     | 
    	
        config.json
    ADDED
    
    | 
         @@ -0,0 +1,25 @@ 
     | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 1 | 
         
            +
            {
         
     | 
| 2 | 
         
            +
              "_name_or_path": "./step_48003",
         
     | 
| 3 | 
         
            +
              "architectures": [
         
     | 
| 4 | 
         
            +
                "BertModel"
         
     | 
| 5 | 
         
            +
              ],
         
     | 
| 6 | 
         
            +
              "attention_probs_dropout_prob": 0.1,
         
     | 
| 7 | 
         
            +
              "classifier_dropout": null,
         
     | 
| 8 | 
         
            +
              "hidden_act": "gelu",
         
     | 
| 9 | 
         
            +
              "hidden_dropout_prob": 0.1,
         
     | 
| 10 | 
         
            +
              "hidden_size": 768,
         
     | 
| 11 | 
         
            +
              "initializer_range": 0.02,
         
     | 
| 12 | 
         
            +
              "intermediate_size": 3072,
         
     | 
| 13 | 
         
            +
              "layer_norm_eps": 1e-12,
         
     | 
| 14 | 
         
            +
              "max_position_embeddings": 512,
         
     | 
| 15 | 
         
            +
              "model_type": "bert",
         
     | 
| 16 | 
         
            +
              "num_attention_heads": 12,
         
     | 
| 17 | 
         
            +
              "num_hidden_layers": 12,
         
     | 
| 18 | 
         
            +
              "pad_token_id": 0,
         
     | 
| 19 | 
         
            +
              "position_embedding_type": "absolute",
         
     | 
| 20 | 
         
            +
              "torch_dtype": "float32",
         
     | 
| 21 | 
         
            +
              "transformers_version": "4.42.4",
         
     | 
| 22 | 
         
            +
              "type_vocab_size": 2,
         
     | 
| 23 | 
         
            +
              "use_cache": true,
         
     | 
| 24 | 
         
            +
              "vocab_size": 32768
         
     | 
| 25 | 
         
            +
            }
         
     | 
    	
        model.safetensors
    ADDED
    
    | 
         @@ -0,0 +1,3 @@ 
     | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 1 | 
         
            +
            version https://git-lfs.github.com/spec/v1
         
     | 
| 2 | 
         
            +
            oid sha256:1d150952429f652c2d2501935e627ebe2f29f89aaf64036dc7eb00b6f2df789a
         
     | 
| 3 | 
         
            +
            size 445245352
         
     | 
    	
        special_tokens_map.json
    ADDED
    
    | 
         @@ -0,0 +1,37 @@ 
     | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 1 | 
         
            +
            {
         
     | 
| 2 | 
         
            +
              "cls_token": {
         
     | 
| 3 | 
         
            +
                "content": "[CLS]",
         
     | 
| 4 | 
         
            +
                "lstrip": false,
         
     | 
| 5 | 
         
            +
                "normalized": false,
         
     | 
| 6 | 
         
            +
                "rstrip": false,
         
     | 
| 7 | 
         
            +
                "single_word": false
         
     | 
| 8 | 
         
            +
              },
         
     | 
| 9 | 
         
            +
              "mask_token": {
         
     | 
| 10 | 
         
            +
                "content": "[MASK]",
         
     | 
| 11 | 
         
            +
                "lstrip": false,
         
     | 
| 12 | 
         
            +
                "normalized": false,
         
     | 
| 13 | 
         
            +
                "rstrip": false,
         
     | 
| 14 | 
         
            +
                "single_word": false
         
     | 
| 15 | 
         
            +
              },
         
     | 
| 16 | 
         
            +
              "pad_token": {
         
     | 
| 17 | 
         
            +
                "content": "[PAD]",
         
     | 
| 18 | 
         
            +
                "lstrip": false,
         
     | 
| 19 | 
         
            +
                "normalized": false,
         
     | 
| 20 | 
         
            +
                "rstrip": false,
         
     | 
| 21 | 
         
            +
                "single_word": false
         
     | 
| 22 | 
         
            +
              },
         
     | 
| 23 | 
         
            +
              "sep_token": {
         
     | 
| 24 | 
         
            +
                "content": "[SEP]",
         
     | 
| 25 | 
         
            +
                "lstrip": false,
         
     | 
| 26 | 
         
            +
                "normalized": false,
         
     | 
| 27 | 
         
            +
                "rstrip": false,
         
     | 
| 28 | 
         
            +
                "single_word": false
         
     | 
| 29 | 
         
            +
              },
         
     | 
| 30 | 
         
            +
              "unk_token": {
         
     | 
| 31 | 
         
            +
                "content": "[UNK]",
         
     | 
| 32 | 
         
            +
                "lstrip": false,
         
     | 
| 33 | 
         
            +
                "normalized": false,
         
     | 
| 34 | 
         
            +
                "rstrip": false,
         
     | 
| 35 | 
         
            +
                "single_word": false
         
     | 
| 36 | 
         
            +
              }
         
     | 
| 37 | 
         
            +
            }
         
     | 
    	
        tokenizer_config.json
    ADDED
    
    | 
         @@ -0,0 +1,63 @@ 
     | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 1 | 
         
            +
            {
         
     | 
| 2 | 
         
            +
              "added_tokens_decoder": {
         
     | 
| 3 | 
         
            +
                "0": {
         
     | 
| 4 | 
         
            +
                  "content": "[PAD]",
         
     | 
| 5 | 
         
            +
                  "lstrip": false,
         
     | 
| 6 | 
         
            +
                  "normalized": false,
         
     | 
| 7 | 
         
            +
                  "rstrip": false,
         
     | 
| 8 | 
         
            +
                  "single_word": false,
         
     | 
| 9 | 
         
            +
                  "special": true
         
     | 
| 10 | 
         
            +
                },
         
     | 
| 11 | 
         
            +
                "1": {
         
     | 
| 12 | 
         
            +
                  "content": "[UNK]",
         
     | 
| 13 | 
         
            +
                  "lstrip": false,
         
     | 
| 14 | 
         
            +
                  "normalized": false,
         
     | 
| 15 | 
         
            +
                  "rstrip": false,
         
     | 
| 16 | 
         
            +
                  "single_word": false,
         
     | 
| 17 | 
         
            +
                  "special": true
         
     | 
| 18 | 
         
            +
                },
         
     | 
| 19 | 
         
            +
                "2": {
         
     | 
| 20 | 
         
            +
                  "content": "[CLS]",
         
     | 
| 21 | 
         
            +
                  "lstrip": false,
         
     | 
| 22 | 
         
            +
                  "normalized": false,
         
     | 
| 23 | 
         
            +
                  "rstrip": false,
         
     | 
| 24 | 
         
            +
                  "single_word": false,
         
     | 
| 25 | 
         
            +
                  "special": true
         
     | 
| 26 | 
         
            +
                },
         
     | 
| 27 | 
         
            +
                "3": {
         
     | 
| 28 | 
         
            +
                  "content": "[SEP]",
         
     | 
| 29 | 
         
            +
                  "lstrip": false,
         
     | 
| 30 | 
         
            +
                  "normalized": false,
         
     | 
| 31 | 
         
            +
                  "rstrip": false,
         
     | 
| 32 | 
         
            +
                  "single_word": false,
         
     | 
| 33 | 
         
            +
                  "special": true
         
     | 
| 34 | 
         
            +
                },
         
     | 
| 35 | 
         
            +
                "4": {
         
     | 
| 36 | 
         
            +
                  "content": "[MASK]",
         
     | 
| 37 | 
         
            +
                  "lstrip": false,
         
     | 
| 38 | 
         
            +
                  "normalized": false,
         
     | 
| 39 | 
         
            +
                  "rstrip": false,
         
     | 
| 40 | 
         
            +
                  "single_word": false,
         
     | 
| 41 | 
         
            +
                  "special": true
         
     | 
| 42 | 
         
            +
                }
         
     | 
| 43 | 
         
            +
              },
         
     | 
| 44 | 
         
            +
              "clean_up_tokenization_spaces": true,
         
     | 
| 45 | 
         
            +
              "cls_token": "[CLS]",
         
     | 
| 46 | 
         
            +
              "do_lower_case": false,
         
     | 
| 47 | 
         
            +
              "do_subword_tokenize": true,
         
     | 
| 48 | 
         
            +
              "do_word_tokenize": true,
         
     | 
| 49 | 
         
            +
              "jumanpp_kwargs": null,
         
     | 
| 50 | 
         
            +
              "mask_token": "[MASK]",
         
     | 
| 51 | 
         
            +
              "mecab_kwargs": {
         
     | 
| 52 | 
         
            +
                "mecab_dic": "unidic_lite"
         
     | 
| 53 | 
         
            +
              },
         
     | 
| 54 | 
         
            +
              "model_max_length": 512,
         
     | 
| 55 | 
         
            +
              "never_split": null,
         
     | 
| 56 | 
         
            +
              "pad_token": "[PAD]",
         
     | 
| 57 | 
         
            +
              "sep_token": "[SEP]",
         
     | 
| 58 | 
         
            +
              "subword_tokenizer_type": "wordpiece",
         
     | 
| 59 | 
         
            +
              "sudachi_kwargs": null,
         
     | 
| 60 | 
         
            +
              "tokenizer_class": "BertJapaneseTokenizer",
         
     | 
| 61 | 
         
            +
              "unk_token": "[UNK]",
         
     | 
| 62 | 
         
            +
              "word_tokenizer_type": "mecab"
         
     | 
| 63 | 
         
            +
            }
         
     | 
    	
        vocab.txt
    ADDED
    
    | 
         The diff for this file is too large to render. 
		See raw diff 
     | 
| 
         |