PeytonT commited on
Commit
caa2c58
·
verified ·
1 Parent(s): fe56dcd

Upload repository_library model package

Browse files
README.md ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: sentence-transformers/all-MiniLM-L6-v2
3
+ library_name: transformers
4
+ pipeline_tag: feature-extraction
5
+ tags:
6
+ - c6
7
+ - cross-modal-retrieval
8
+ - repository-library
9
+ - research-library
10
+ - retrieval
11
+ - t5_cross
12
+ ---
13
+
14
+ # Cross Modal Retrieval
15
+
16
+ Retrieves semantically related items across paper and repository modalities.
17
+
18
+ ## Model Details
19
+
20
+ - Artifact type: full fine-tuned model
21
+ - Base model: `sentence-transformers/all-MiniLM-L6-v2`
22
+ - Model ID: `C6`
23
+ - Tier: `T5_cross`
24
+ - Local mirror: `/arxiv/models/repository_library/cross-modal-retrieval`
25
+ - Source checkpoint: `models/checkpoints/C6`
26
+
27
+ This repository is part of the `repository_library` model stack and is mirrored from `/data/repository_library/models/checkpoints` for publication under the `PeytonT` namespace.
28
+
29
+ ## Intended Use
30
+
31
+ - Primary use: Retrieves semantically related items across paper and repository modalities.
32
+ - Secondary use: retrieval, ranking, planning, or scientific paper tooling inside the broader Repository Library system, depending on the model family.
33
+ - Out of scope: production safety claims, benchmark claims beyond the bundled experiment config, or use outside the model's narrow training objective without task-specific validation.
34
+
35
+ ## Training Data
36
+
37
+ This package was trained from the following declared datasets or corpus sources:
38
+
39
+ - `source:arxiv_pdfs_structured`
40
+ - `source:github_repos`
41
+
42
+ ## Training Procedure
43
+
44
+ - Sources: `arxiv_pdfs_structured, github_repos`
45
+ - Input fields: `paper_chunk, code_chunk`
46
+ - Target fields: `shared_embedding`
47
+ - Max samples: `4000`
48
+ - Precision: `bf16`
49
+ - Objective: `contrastive`
50
+ - Batch size: `8`
51
+ - Learning rate: `5e-05`
52
+ - Max source tokens: `256`
53
+ - Max target tokens: `128`
54
+ - Max steps: `1000`
55
+
56
+ ## Evaluation
57
+
58
+ - Declared metrics: `recall_at_10, ndcg_at_10`
59
+ - Status: local experiment artifact mirrored for release; external benchmark reporting has not been standardized across the full model family yet.
60
+
61
+ ## Usage
62
+
63
+ ```python
64
+ from transformers import AutoModel, AutoTokenizer
65
+
66
+ repo_id = "PeytonT/cross-modal-retrieval"
67
+
68
+ tokenizer = AutoTokenizer.from_pretrained(repo_id)
69
+ model = AutoModel.from_pretrained(repo_id)
70
+ ```
71
+
72
+ ## Limitations
73
+
74
+ - These model cards reflect the packaged experiment configs and mirrored checkpoint contents, not an independently audited benchmark sheet.
75
+ - Some training datasets are local corpora or exported shards, so reproducibility may require access to the surrounding Repository Library data pipeline.
76
+ - Models in this stack are narrow components of a larger paper-and-repository system and should be validated on downstream tasks before deployment.
77
+
78
+ ## Project Context
79
+
80
+ Repository Library is a research system for indexing, retrieving, aligning, and reasoning over scientific papers, structured paper content, repositories, and cross-domain links between them.
81
+
82
+ ## Contact
83
+
84
+ Published under `PeytonT` from the local `repository_library` build.
config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertModel"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "dtype": "float32",
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 384,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 1536,
14
+ "layer_norm_eps": 1e-12,
15
+ "max_position_embeddings": 512,
16
+ "model_type": "bert",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 6,
19
+ "pad_token_id": 0,
20
+ "position_embedding_type": "absolute",
21
+ "transformers_version": "4.57.6",
22
+ "type_vocab_size": 2,
23
+ "use_cache": true,
24
+ "vocab_size": 30522
25
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b626e987ce893ecb2c06adf82bb6ff5355ce227de1ae47006a7c9a421827b088
3
+ size 90864192
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.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
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": false,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "max_length": 128,
51
+ "model_max_length": 512,
52
+ "never_split": null,
53
+ "pad_to_multiple_of": null,
54
+ "pad_token": "[PAD]",
55
+ "pad_token_type_id": 0,
56
+ "padding_side": "right",
57
+ "sep_token": "[SEP]",
58
+ "stride": 0,
59
+ "strip_accents": null,
60
+ "tokenize_chinese_chars": true,
61
+ "tokenizer_class": "BertTokenizer",
62
+ "truncation_side": "right",
63
+ "truncation_strategy": "longest_first",
64
+ "unk_token": "[UNK]"
65
+ }
trainer_state.json ADDED
@@ -0,0 +1,183 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 40.0,
6
+ "eval_steps": 500,
7
+ "global_step": 1000,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 2.0,
14
+ "grad_norm": 1.2906718254089355,
15
+ "learning_rate": 2.45e-05,
16
+ "loss": 0.3942,
17
+ "step": 50
18
+ },
19
+ {
20
+ "epoch": 4.0,
21
+ "grad_norm": 2.3395345211029053,
22
+ "learning_rate": 4.9500000000000004e-05,
23
+ "loss": 0.3539,
24
+ "step": 100
25
+ },
26
+ {
27
+ "epoch": 6.0,
28
+ "grad_norm": 1.5900944471359253,
29
+ "learning_rate": 4.727777777777778e-05,
30
+ "loss": 0.2985,
31
+ "step": 150
32
+ },
33
+ {
34
+ "epoch": 8.0,
35
+ "grad_norm": 1.4815367460250854,
36
+ "learning_rate": 4.4500000000000004e-05,
37
+ "loss": 0.2351,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 10.0,
42
+ "grad_norm": 2.1699378490448,
43
+ "learning_rate": 4.172222222222222e-05,
44
+ "loss": 0.1905,
45
+ "step": 250
46
+ },
47
+ {
48
+ "epoch": 12.0,
49
+ "grad_norm": 3.293229341506958,
50
+ "learning_rate": 3.894444444444444e-05,
51
+ "loss": 0.1523,
52
+ "step": 300
53
+ },
54
+ {
55
+ "epoch": 14.0,
56
+ "grad_norm": 2.814497232437134,
57
+ "learning_rate": 3.6166666666666674e-05,
58
+ "loss": 0.1249,
59
+ "step": 350
60
+ },
61
+ {
62
+ "epoch": 16.0,
63
+ "grad_norm": 1.806630253791809,
64
+ "learning_rate": 3.338888888888889e-05,
65
+ "loss": 0.1036,
66
+ "step": 400
67
+ },
68
+ {
69
+ "epoch": 18.0,
70
+ "grad_norm": 1.9773627519607544,
71
+ "learning_rate": 3.061111111111111e-05,
72
+ "loss": 0.0848,
73
+ "step": 450
74
+ },
75
+ {
76
+ "epoch": 20.0,
77
+ "grad_norm": 2.187488079071045,
78
+ "learning_rate": 2.7833333333333333e-05,
79
+ "loss": 0.0745,
80
+ "step": 500
81
+ },
82
+ {
83
+ "epoch": 22.0,
84
+ "grad_norm": 1.9318913221359253,
85
+ "learning_rate": 2.5055555555555555e-05,
86
+ "loss": 0.0628,
87
+ "step": 550
88
+ },
89
+ {
90
+ "epoch": 24.0,
91
+ "grad_norm": 1.486013412475586,
92
+ "learning_rate": 2.2277777777777778e-05,
93
+ "loss": 0.0573,
94
+ "step": 600
95
+ },
96
+ {
97
+ "epoch": 26.0,
98
+ "grad_norm": 0.9214858412742615,
99
+ "learning_rate": 1.9500000000000003e-05,
100
+ "loss": 0.049,
101
+ "step": 650
102
+ },
103
+ {
104
+ "epoch": 28.0,
105
+ "grad_norm": 2.187666177749634,
106
+ "learning_rate": 1.6722222222222222e-05,
107
+ "loss": 0.0475,
108
+ "step": 700
109
+ },
110
+ {
111
+ "epoch": 30.0,
112
+ "grad_norm": 1.7305703163146973,
113
+ "learning_rate": 1.3944444444444446e-05,
114
+ "loss": 0.0424,
115
+ "step": 750
116
+ },
117
+ {
118
+ "epoch": 32.0,
119
+ "grad_norm": 1.5193572044372559,
120
+ "learning_rate": 1.1166666666666668e-05,
121
+ "loss": 0.0392,
122
+ "step": 800
123
+ },
124
+ {
125
+ "epoch": 34.0,
126
+ "grad_norm": 1.132284164428711,
127
+ "learning_rate": 8.38888888888889e-06,
128
+ "loss": 0.0365,
129
+ "step": 850
130
+ },
131
+ {
132
+ "epoch": 36.0,
133
+ "grad_norm": 0.8349636793136597,
134
+ "learning_rate": 5.611111111111112e-06,
135
+ "loss": 0.0353,
136
+ "step": 900
137
+ },
138
+ {
139
+ "epoch": 38.0,
140
+ "grad_norm": 0.5874630808830261,
141
+ "learning_rate": 2.8333333333333335e-06,
142
+ "loss": 0.0337,
143
+ "step": 950
144
+ },
145
+ {
146
+ "epoch": 40.0,
147
+ "grad_norm": 0.44510239362716675,
148
+ "learning_rate": 5.555555555555556e-08,
149
+ "loss": 0.033,
150
+ "step": 1000
151
+ },
152
+ {
153
+ "epoch": 40.0,
154
+ "step": 1000,
155
+ "total_flos": 0.0,
156
+ "train_loss": 0.12244693279266357,
157
+ "train_runtime": 147.0867,
158
+ "train_samples_per_second": 163.169,
159
+ "train_steps_per_second": 6.799
160
+ }
161
+ ],
162
+ "logging_steps": 50,
163
+ "max_steps": 1000,
164
+ "num_input_tokens_seen": 0,
165
+ "num_train_epochs": 40,
166
+ "save_steps": 200,
167
+ "stateful_callbacks": {
168
+ "TrainerControl": {
169
+ "args": {
170
+ "should_epoch_stop": false,
171
+ "should_evaluate": false,
172
+ "should_log": false,
173
+ "should_save": true,
174
+ "should_training_stop": true
175
+ },
176
+ "attributes": {}
177
+ }
178
+ },
179
+ "total_flos": 0.0,
180
+ "train_batch_size": 24,
181
+ "trial_name": null,
182
+ "trial_params": null
183
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b19986ddf3e1a8b66f7011e90c87f0be58a51028eae27a0aebf7588c9d4323b
3
+ size 5841
vocab.txt ADDED
The diff for this file is too large to render. See raw diff