joshuaclymer commited on
Commit
a5bd7bf
·
1 Parent(s): ef11752

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ ---
4
+ ## Training procedure
5
+
6
+
7
+ The following `bitsandbytes` quantization config was used during training:
8
+ - quant_method: bitsandbytes
9
+ - load_in_8bit: False
10
+ - load_in_4bit: True
11
+ - llm_int8_threshold: 6.0
12
+ - llm_int8_skip_modules: None
13
+ - llm_int8_enable_fp32_cpu_offload: False
14
+ - llm_int8_has_fp16_weight: False
15
+ - bnb_4bit_quant_type: nf4
16
+ - bnb_4bit_use_double_quant: True
17
+ - bnb_4bit_compute_dtype: float16
18
+ ### Framework versions
19
+
20
+
21
+ - PEFT 0.5.0
adapter_config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "auto_mapping": null,
3
+ "base_model_name_or_path": "models/openllama-3b",
4
+ "bias": "none",
5
+ "fan_in_fan_out": false,
6
+ "inference_mode": true,
7
+ "init_lora_weights": true,
8
+ "layers_pattern": null,
9
+ "layers_to_transform": null,
10
+ "lora_alpha": 16,
11
+ "lora_dropout": 0.1,
12
+ "modules_to_save": null,
13
+ "peft_type": "LORA",
14
+ "r": 64,
15
+ "revision": null,
16
+ "target_modules": [
17
+ "q_proj",
18
+ "v_proj"
19
+ ],
20
+ "task_type": "SEQ_CLS"
21
+ }
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9230bc7e5e060e3bb3cebd4225869e5c7f5b009b3bf2adfb1269274c45f66ed
3
+ size 85258545
special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "</s>",
17
+ "unk_token": {
18
+ "content": "<unk>",
19
+ "lstrip": false,
20
+ "normalized": true,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab1b681ec7fc02fed5edd3026687d7a692a918c4dd8e150ca2e3994a6229843b
3
+ size 534194
tokenizer_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "__type": "AddedToken",
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": true,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ "clean_up_tokenization_spaces": false,
11
+ "eos_token": {
12
+ "__type": "AddedToken",
13
+ "content": "</s>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "legacy": null,
20
+ "model_max_length": 2048,
21
+ "pad_token": null,
22
+ "sp_model_kwargs": {},
23
+ "spaces_between_special_tokens": false,
24
+ "tokenizer_class": "LlamaTokenizer",
25
+ "unk_token": {
26
+ "__type": "AddedToken",
27
+ "content": "<unk>",
28
+ "lstrip": false,
29
+ "normalized": true,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ },
33
+ "use_default_system_prompt": true
34
+ }
train_args.json ADDED
@@ -0,0 +1,123 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "initial_model_dir": "models/openllama-3b",
3
+ "distribution_id": "word_swap",
4
+ "date_trained": "15/10/2023 09:10:37",
5
+ "output_dir": "models/classify_lora/openllama-3b-word_swap",
6
+ "overwrite_output_dir": false,
7
+ "do_train": false,
8
+ "do_eval": true,
9
+ "do_predict": false,
10
+ "evaluation_strategy": "steps",
11
+ "prediction_loss_only": false,
12
+ "per_device_train_batch_size": 16,
13
+ "per_device_eval_batch_size": 32,
14
+ "per_gpu_train_batch_size": null,
15
+ "per_gpu_eval_batch_size": null,
16
+ "gradient_accumulation_steps": 1,
17
+ "eval_accumulation_steps": null,
18
+ "eval_delay": 0,
19
+ "learning_rate": 7.2e-05,
20
+ "weight_decay": 0.0,
21
+ "adam_beta1": 0.9,
22
+ "adam_beta2": 0.999,
23
+ "adam_epsilon": 1e-08,
24
+ "max_grad_norm": 0.3,
25
+ "num_train_epochs": 2,
26
+ "max_steps": 100,
27
+ "lr_scheduler_type": "constant",
28
+ "warmup_ratio": 0.03,
29
+ "warmup_steps": 0,
30
+ "log_level": "passive",
31
+ "log_level_replica": "warning",
32
+ "log_on_each_node": true,
33
+ "logging_dir": "models/classify_lora/openllama-3b-word_swap/runs/Oct15_08-56-43_compute-permanent-node-978",
34
+ "logging_strategy": "steps",
35
+ "logging_first_step": false,
36
+ "logging_steps": 1,
37
+ "logging_nan_inf_filter": true,
38
+ "save_strategy": "steps",
39
+ "save_steps": 25,
40
+ "save_total_limit": 0,
41
+ "save_safetensors": false,
42
+ "save_on_each_node": false,
43
+ "no_cuda": false,
44
+ "use_cpu": false,
45
+ "use_mps_device": false,
46
+ "seed": 42,
47
+ "data_seed": null,
48
+ "jit_mode_eval": false,
49
+ "use_ipex": false,
50
+ "bf16": false,
51
+ "fp16": false,
52
+ "fp16_opt_level": "O1",
53
+ "half_precision_backend": "auto",
54
+ "bf16_full_eval": false,
55
+ "fp16_full_eval": false,
56
+ "tf32": null,
57
+ "local_rank": 0,
58
+ "ddp_backend": null,
59
+ "tpu_num_cores": null,
60
+ "tpu_metrics_debug": false,
61
+ "debug": [],
62
+ "dataloader_drop_last": false,
63
+ "eval_steps": 25,
64
+ "dataloader_num_workers": 0,
65
+ "past_index": -1,
66
+ "run_name": "train|models-classify_lora-openllama-3b-word_swap",
67
+ "disable_tqdm": false,
68
+ "remove_unused_columns": false,
69
+ "label_names": null,
70
+ "load_best_model_at_end": false,
71
+ "metric_for_best_model": "eval_word_swap_score",
72
+ "greater_is_better": true,
73
+ "ignore_data_skip": false,
74
+ "sharded_ddp": [],
75
+ "fsdp": [],
76
+ "fsdp_min_num_params": 0,
77
+ "fsdp_config": {
78
+ "min_num_params": 0,
79
+ "xla": false,
80
+ "xla_fsdp_grad_ckpt": false
81
+ },
82
+ "fsdp_transformer_layer_cls_to_wrap": null,
83
+ "deepspeed": "configs/ds_zero_1.json",
84
+ "label_smoothing_factor": 0.0,
85
+ "optim": "paged_adamw_32bit",
86
+ "optim_args": null,
87
+ "adafactor": false,
88
+ "group_by_length": false,
89
+ "length_column_name": "length",
90
+ "report_to": [
91
+ "wandb"
92
+ ],
93
+ "ddp_find_unused_parameters": false,
94
+ "ddp_bucket_cap_mb": null,
95
+ "ddp_broadcast_buffers": null,
96
+ "dataloader_pin_memory": true,
97
+ "skip_memory_metrics": true,
98
+ "use_legacy_prediction_loop": false,
99
+ "push_to_hub": false,
100
+ "resume_from_checkpoint": null,
101
+ "hub_model_id": null,
102
+ "hub_strategy": "every_save",
103
+ "hub_token": null,
104
+ "hub_private_repo": false,
105
+ "hub_always_push": false,
106
+ "gradient_checkpointing": false,
107
+ "include_inputs_for_metrics": false,
108
+ "fp16_backend": "auto",
109
+ "push_to_hub_model_id": null,
110
+ "push_to_hub_organization": null,
111
+ "push_to_hub_token": null,
112
+ "_n_gpu": 1,
113
+ "mp_parameters": "",
114
+ "auto_find_batch_size": false,
115
+ "full_determinism": false,
116
+ "torchdynamo": null,
117
+ "ray_scope": "last",
118
+ "ddp_timeout": 1800,
119
+ "torch_compile": false,
120
+ "torch_compile_backend": null,
121
+ "torch_compile_mode": null,
122
+ "dispatch_batches": null
123
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb2da315a588212d197439c16ff81d474f2368fc6e1c6475681eb63ba38d4d9f
3
+ size 5115
training_logs.json ADDED
@@ -0,0 +1,1067 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "loss": 0.686,
4
+ "learning_rate": 7.2e-05,
5
+ "epoch": 0.02,
6
+ "step": 1
7
+ },
8
+ {
9
+ "loss": 0.6888,
10
+ "learning_rate": 7.2e-05,
11
+ "epoch": 0.04,
12
+ "step": 2
13
+ },
14
+ {
15
+ "loss": 0.6939,
16
+ "learning_rate": 7.2e-05,
17
+ "epoch": 0.05,
18
+ "step": 3
19
+ },
20
+ {
21
+ "loss": 0.6792,
22
+ "learning_rate": 7.2e-05,
23
+ "epoch": 0.07,
24
+ "step": 4
25
+ },
26
+ {
27
+ "loss": 0.6878,
28
+ "learning_rate": 7.2e-05,
29
+ "epoch": 0.09,
30
+ "step": 5
31
+ },
32
+ {
33
+ "loss": 0.7054,
34
+ "learning_rate": 7.2e-05,
35
+ "epoch": 0.11,
36
+ "step": 6
37
+ },
38
+ {
39
+ "loss": 0.6631,
40
+ "learning_rate": 7.2e-05,
41
+ "epoch": 0.12,
42
+ "step": 7
43
+ },
44
+ {
45
+ "loss": 0.6819,
46
+ "learning_rate": 7.2e-05,
47
+ "epoch": 0.14,
48
+ "step": 8
49
+ },
50
+ {
51
+ "loss": 0.6589,
52
+ "learning_rate": 7.2e-05,
53
+ "epoch": 0.16,
54
+ "step": 9
55
+ },
56
+ {
57
+ "loss": 0.6671,
58
+ "learning_rate": 7.2e-05,
59
+ "epoch": 0.18,
60
+ "step": 10
61
+ },
62
+ {
63
+ "loss": 0.6703,
64
+ "learning_rate": 7.2e-05,
65
+ "epoch": 0.19,
66
+ "step": 11
67
+ },
68
+ {
69
+ "loss": 0.6868,
70
+ "learning_rate": 7.2e-05,
71
+ "epoch": 0.21,
72
+ "step": 12
73
+ },
74
+ {
75
+ "loss": 0.6695,
76
+ "learning_rate": 7.2e-05,
77
+ "epoch": 0.23,
78
+ "step": 13
79
+ },
80
+ {
81
+ "loss": 0.6697,
82
+ "learning_rate": 7.2e-05,
83
+ "epoch": 0.25,
84
+ "step": 14
85
+ },
86
+ {
87
+ "loss": 0.6752,
88
+ "learning_rate": 7.2e-05,
89
+ "epoch": 0.26,
90
+ "step": 15
91
+ },
92
+ {
93
+ "loss": 0.655,
94
+ "learning_rate": 7.2e-05,
95
+ "epoch": 0.28,
96
+ "step": 16
97
+ },
98
+ {
99
+ "loss": 0.6622,
100
+ "learning_rate": 7.2e-05,
101
+ "epoch": 0.3,
102
+ "step": 17
103
+ },
104
+ {
105
+ "loss": 0.6451,
106
+ "learning_rate": 7.2e-05,
107
+ "epoch": 0.32,
108
+ "step": 18
109
+ },
110
+ {
111
+ "loss": 0.6529,
112
+ "learning_rate": 7.2e-05,
113
+ "epoch": 0.33,
114
+ "step": 19
115
+ },
116
+ {
117
+ "loss": 0.6682,
118
+ "learning_rate": 7.2e-05,
119
+ "epoch": 0.35,
120
+ "step": 20
121
+ },
122
+ {
123
+ "loss": 0.6333,
124
+ "learning_rate": 7.2e-05,
125
+ "epoch": 0.37,
126
+ "step": 21
127
+ },
128
+ {
129
+ "loss": 0.6474,
130
+ "learning_rate": 7.2e-05,
131
+ "epoch": 0.39,
132
+ "step": 22
133
+ },
134
+ {
135
+ "loss": 0.6165,
136
+ "learning_rate": 7.2e-05,
137
+ "epoch": 0.4,
138
+ "step": 23
139
+ },
140
+ {
141
+ "loss": 0.6007,
142
+ "learning_rate": 7.2e-05,
143
+ "epoch": 0.42,
144
+ "step": 24
145
+ },
146
+ {
147
+ "loss": 0.6468,
148
+ "learning_rate": 7.2e-05,
149
+ "epoch": 0.44,
150
+ "step": 25
151
+ },
152
+ {
153
+ "eval_word_swap_loss": 0.6271592974662781,
154
+ "eval_word_swap_score": -0.21765999495983124,
155
+ "eval_word_swap_brier_score": 0.21765999495983124,
156
+ "eval_word_swap_average_probability": 0.5386337041854858,
157
+ "eval_word_swap_accuracy": 0.74,
158
+ "eval_word_swap_probabilities": [
159
+ 0.5120093822479248,
160
+ 0.5243922472000122,
161
+ 0.5167696475982666,
162
+ 0.5557374358177185,
163
+ 0.5420051217079163,
164
+ 0.5566287040710449,
165
+ 0.4828156530857086,
166
+ 0.48950111865997314,
167
+ 0.4811334013938904,
168
+ 0.4968768358230591,
169
+ 0.5213818550109863,
170
+ 0.4816601872444153,
171
+ 0.4400249719619751,
172
+ 0.45116183161735535,
173
+ 0.5269880294799805,
174
+ 0.7658145427703857,
175
+ 0.7190629839897156,
176
+ 0.777766764163971,
177
+ 0.5307355523109436,
178
+ 0.5157463550567627,
179
+ 0.4815860390663147,
180
+ 0.6414819359779358,
181
+ 0.6972522139549255,
182
+ 0.6069620847702026,
183
+ 0.49553626775741577,
184
+ 0.5025679469108582,
185
+ 0.5022831559181213,
186
+ 0.4942616820335388,
187
+ 0.49401095509529114,
188
+ 0.49113011360168457,
189
+ 0.5141912698745728,
190
+ 0.4628845155239105,
191
+ 0.4878145456314087,
192
+ 0.4648214280605316,
193
+ 0.49921849370002747,
194
+ 0.5039854049682617,
195
+ 0.516721785068512,
196
+ 0.5200318098068237,
197
+ 0.5145192742347717,
198
+ 0.5270195603370667,
199
+ 0.49075567722320557,
200
+ 0.5064305663108826,
201
+ 0.5328724980354309,
202
+ 0.5033775568008423,
203
+ 0.4973742663860321,
204
+ 0.7190636396408081,
205
+ 0.7096593379974365,
206
+ 0.7158627510070801,
207
+ 0.4918597638607025,
208
+ 0.5082303285598755,
209
+ 0.5219036340713501,
210
+ 0.5261024832725525,
211
+ 0.5440922975540161,
212
+ 0.515591561794281,
213
+ 0.5325610637664795,
214
+ 0.5472833514213562,
215
+ 0.5047705769538879,
216
+ 0.6283226609230042,
217
+ 0.6890944838523865,
218
+ 0.6690213084220886,
219
+ 0.5016189813613892,
220
+ 0.4952182471752167,
221
+ 0.5048976540565491,
222
+ 0.5249401926994324,
223
+ 0.5131137371063232,
224
+ 0.4067339599132538,
225
+ 0.5245459079742432,
226
+ 0.5194306969642639,
227
+ 0.5493549704551697,
228
+ 0.5752891302108765,
229
+ 0.5456081032752991,
230
+ 0.6180267333984375,
231
+ 0.5227917432785034,
232
+ 0.47111740708351135,
233
+ 0.5111400485038757,
234
+ 0.5506094694137573,
235
+ 0.5049382448196411,
236
+ 0.5248739123344421,
237
+ 0.49835801124572754,
238
+ 0.4988904297351837,
239
+ 0.5008316040039062,
240
+ 0.5080080628395081,
241
+ 0.5087270140647888,
242
+ 0.5232927203178406,
243
+ 0.5299203395843506,
244
+ 0.548637330532074,
245
+ 0.6203308701515198,
246
+ 0.6104732155799866,
247
+ 0.6011025905609131,
248
+ 0.6144075989723206,
249
+ 0.5003787279129028,
250
+ 0.5005776286125183,
251
+ 0.4993947148323059,
252
+ 0.5284203886985779,
253
+ 0.5499283671379089,
254
+ 0.5445973873138428,
255
+ 0.5152997970581055,
256
+ 0.6039677858352661,
257
+ 0.5481812953948975,
258
+ 0.4846700429916382
259
+ ],
260
+ "eval_word_swap_runtime": 10.8997,
261
+ "eval_word_swap_samples_per_second": 9.175,
262
+ "eval_word_swap_steps_per_second": 0.183,
263
+ "epoch": 0.44,
264
+ "step": 25
265
+ },
266
+ {
267
+ "loss": 0.619,
268
+ "learning_rate": 7.2e-05,
269
+ "epoch": 0.46,
270
+ "step": 26
271
+ },
272
+ {
273
+ "loss": 0.5437,
274
+ "learning_rate": 7.2e-05,
275
+ "epoch": 0.47,
276
+ "step": 27
277
+ },
278
+ {
279
+ "loss": 0.5786,
280
+ "learning_rate": 7.2e-05,
281
+ "epoch": 0.49,
282
+ "step": 28
283
+ },
284
+ {
285
+ "loss": 0.5315,
286
+ "learning_rate": 7.2e-05,
287
+ "epoch": 0.51,
288
+ "step": 29
289
+ },
290
+ {
291
+ "loss": 0.6166,
292
+ "learning_rate": 7.2e-05,
293
+ "epoch": 0.53,
294
+ "step": 30
295
+ },
296
+ {
297
+ "loss": 0.4868,
298
+ "learning_rate": 7.2e-05,
299
+ "epoch": 0.54,
300
+ "step": 31
301
+ },
302
+ {
303
+ "loss": 0.535,
304
+ "learning_rate": 7.2e-05,
305
+ "epoch": 0.56,
306
+ "step": 32
307
+ },
308
+ {
309
+ "loss": 0.569,
310
+ "learning_rate": 7.2e-05,
311
+ "epoch": 0.58,
312
+ "step": 33
313
+ },
314
+ {
315
+ "loss": 0.5411,
316
+ "learning_rate": 7.2e-05,
317
+ "epoch": 0.6,
318
+ "step": 34
319
+ },
320
+ {
321
+ "loss": 0.5294,
322
+ "learning_rate": 7.2e-05,
323
+ "epoch": 0.61,
324
+ "step": 35
325
+ },
326
+ {
327
+ "loss": 0.4853,
328
+ "learning_rate": 7.2e-05,
329
+ "epoch": 0.63,
330
+ "step": 36
331
+ },
332
+ {
333
+ "loss": 0.391,
334
+ "learning_rate": 7.2e-05,
335
+ "epoch": 0.65,
336
+ "step": 37
337
+ },
338
+ {
339
+ "loss": 0.4294,
340
+ "learning_rate": 7.2e-05,
341
+ "epoch": 0.67,
342
+ "step": 38
343
+ },
344
+ {
345
+ "loss": 0.4508,
346
+ "learning_rate": 7.2e-05,
347
+ "epoch": 0.68,
348
+ "step": 39
349
+ },
350
+ {
351
+ "loss": 0.4035,
352
+ "learning_rate": 7.2e-05,
353
+ "epoch": 0.7,
354
+ "step": 40
355
+ },
356
+ {
357
+ "loss": 0.3999,
358
+ "learning_rate": 7.2e-05,
359
+ "epoch": 0.72,
360
+ "step": 41
361
+ },
362
+ {
363
+ "loss": 0.4833,
364
+ "learning_rate": 7.2e-05,
365
+ "epoch": 0.74,
366
+ "step": 42
367
+ },
368
+ {
369
+ "loss": 0.2988,
370
+ "learning_rate": 7.2e-05,
371
+ "epoch": 0.75,
372
+ "step": 43
373
+ },
374
+ {
375
+ "loss": 0.2764,
376
+ "learning_rate": 7.2e-05,
377
+ "epoch": 0.77,
378
+ "step": 44
379
+ },
380
+ {
381
+ "loss": 0.2748,
382
+ "learning_rate": 7.2e-05,
383
+ "epoch": 0.79,
384
+ "step": 45
385
+ },
386
+ {
387
+ "loss": 0.5379,
388
+ "learning_rate": 7.2e-05,
389
+ "epoch": 0.81,
390
+ "step": 46
391
+ },
392
+ {
393
+ "loss": 0.3387,
394
+ "learning_rate": 7.2e-05,
395
+ "epoch": 0.82,
396
+ "step": 47
397
+ },
398
+ {
399
+ "loss": 0.3761,
400
+ "learning_rate": 7.2e-05,
401
+ "epoch": 0.84,
402
+ "step": 48
403
+ },
404
+ {
405
+ "loss": 0.3949,
406
+ "learning_rate": 7.2e-05,
407
+ "epoch": 0.86,
408
+ "step": 49
409
+ },
410
+ {
411
+ "loss": 0.3446,
412
+ "learning_rate": 7.2e-05,
413
+ "epoch": 0.88,
414
+ "step": 50
415
+ },
416
+ {
417
+ "eval_word_swap_loss": 0.42554885149002075,
418
+ "eval_word_swap_score": -0.140316903591156,
419
+ "eval_word_swap_brier_score": 0.140316903591156,
420
+ "eval_word_swap_average_probability": 0.7100909352302551,
421
+ "eval_word_swap_accuracy": 0.79,
422
+ "eval_word_swap_probabilities": [
423
+ 0.4019075036048889,
424
+ 0.37677139043807983,
425
+ 0.4804254174232483,
426
+ 0.9920067191123962,
427
+ 0.9952875375747681,
428
+ 0.9941473007202148,
429
+ 0.5543147325515747,
430
+ 0.38835442066192627,
431
+ 0.3292403817176819,
432
+ 0.6027093529701233,
433
+ 0.5990132689476013,
434
+ 0.7515374422073364,
435
+ 0.4571269750595093,
436
+ 0.48707351088523865,
437
+ 0.5104720592498779,
438
+ 0.9999973773956299,
439
+ 0.998894989490509,
440
+ 0.9999964237213135,
441
+ 0.7822933197021484,
442
+ 0.80682772397995,
443
+ 0.7521791458129883,
444
+ 0.9994971752166748,
445
+ 0.9999256134033203,
446
+ 0.9836665391921997,
447
+ 0.576387345790863,
448
+ 0.6108314394950867,
449
+ 0.6564780473709106,
450
+ 0.49568498134613037,
451
+ 0.48642292618751526,
452
+ 0.5067446827888489,
453
+ 0.613825261592865,
454
+ 0.4119172692298889,
455
+ 0.6788980960845947,
456
+ 0.4296410083770752,
457
+ 0.438064843416214,
458
+ 0.8100086450576782,
459
+ 0.4851573407649994,
460
+ 0.5169422030448914,
461
+ 0.49324727058410645,
462
+ 0.49987977743148804,
463
+ 0.33004453778266907,
464
+ 0.5734589099884033,
465
+ 0.7685950994491577,
466
+ 0.9241077899932861,
467
+ 0.6195946931838989,
468
+ 0.9778075218200684,
469
+ 0.9802026748657227,
470
+ 0.9920069575309753,
471
+ 0.48942628502845764,
472
+ 0.6928991675376892,
473
+ 0.7374966144561768,
474
+ 0.5482223629951477,
475
+ 0.9530254602432251,
476
+ 0.39021438360214233,
477
+ 0.6551928520202637,
478
+ 0.956979513168335,
479
+ 0.3156083822250366,
480
+ 0.9999629259109497,
481
+ 0.9999154806137085,
482
+ 0.9999653100967407,
483
+ 0.5982706546783447,
484
+ 0.6141578555107117,
485
+ 0.7660197615623474,
486
+ 0.5191519856452942,
487
+ 0.19016587734222412,
488
+ 0.08200426399707794,
489
+ 0.996311366558075,
490
+ 0.8929523825645447,
491
+ 0.9970592260360718,
492
+ 0.9973663687705994,
493
+ 0.9897236824035645,
494
+ 0.9978641867637634,
495
+ 0.5952801704406738,
496
+ 0.673331081867218,
497
+ 0.6558879017829895,
498
+ 0.6988264322280884,
499
+ 0.5485352277755737,
500
+ 0.5909713506698608,
501
+ 0.5130484700202942,
502
+ 0.5165088176727295,
503
+ 0.5258926749229431,
504
+ 0.8345712423324585,
505
+ 0.6373376846313477,
506
+ 0.8837608098983765,
507
+ 0.9874070286750793,
508
+ 0.9953998923301697,
509
+ 0.9999574422836304,
510
+ 0.9886431097984314,
511
+ 0.9805268049240112,
512
+ 0.9796808958053589,
513
+ 0.5108579993247986,
514
+ 0.5106061100959778,
515
+ 0.5018303394317627,
516
+ 0.8787161111831665,
517
+ 0.9941256046295166,
518
+ 0.9769755601882935,
519
+ 0.99406898021698,
520
+ 0.9958688020706177,
521
+ 0.9664839506149292,
522
+ 0.5744118094444275
523
+ ],
524
+ "eval_word_swap_runtime": 10.8934,
525
+ "eval_word_swap_samples_per_second": 9.18,
526
+ "eval_word_swap_steps_per_second": 0.184,
527
+ "epoch": 0.88,
528
+ "step": 50
529
+ },
530
+ {
531
+ "loss": 0.3499,
532
+ "learning_rate": 7.2e-05,
533
+ "epoch": 0.89,
534
+ "step": 51
535
+ },
536
+ {
537
+ "loss": 0.4856,
538
+ "learning_rate": 7.2e-05,
539
+ "epoch": 0.91,
540
+ "step": 52
541
+ },
542
+ {
543
+ "loss": 0.4315,
544
+ "learning_rate": 7.2e-05,
545
+ "epoch": 0.93,
546
+ "step": 53
547
+ },
548
+ {
549
+ "loss": 0.192,
550
+ "learning_rate": 7.2e-05,
551
+ "epoch": 0.95,
552
+ "step": 54
553
+ },
554
+ {
555
+ "loss": 0.2681,
556
+ "learning_rate": 7.2e-05,
557
+ "epoch": 0.96,
558
+ "step": 55
559
+ },
560
+ {
561
+ "loss": 0.4028,
562
+ "learning_rate": 7.2e-05,
563
+ "epoch": 0.98,
564
+ "step": 56
565
+ },
566
+ {
567
+ "loss": 0.2514,
568
+ "learning_rate": 7.2e-05,
569
+ "epoch": 1.0,
570
+ "step": 57
571
+ },
572
+ {
573
+ "loss": 0.2267,
574
+ "learning_rate": 7.2e-05,
575
+ "epoch": 1.02,
576
+ "step": 58
577
+ },
578
+ {
579
+ "loss": 0.2351,
580
+ "learning_rate": 7.2e-05,
581
+ "epoch": 1.04,
582
+ "step": 59
583
+ },
584
+ {
585
+ "loss": 0.2467,
586
+ "learning_rate": 7.2e-05,
587
+ "epoch": 1.05,
588
+ "step": 60
589
+ },
590
+ {
591
+ "loss": 0.3004,
592
+ "learning_rate": 7.2e-05,
593
+ "epoch": 1.07,
594
+ "step": 61
595
+ },
596
+ {
597
+ "loss": 0.217,
598
+ "learning_rate": 7.2e-05,
599
+ "epoch": 1.09,
600
+ "step": 62
601
+ },
602
+ {
603
+ "loss": 0.2547,
604
+ "learning_rate": 7.2e-05,
605
+ "epoch": 1.11,
606
+ "step": 63
607
+ },
608
+ {
609
+ "loss": 0.3135,
610
+ "learning_rate": 7.2e-05,
611
+ "epoch": 1.12,
612
+ "step": 64
613
+ },
614
+ {
615
+ "loss": 0.242,
616
+ "learning_rate": 7.2e-05,
617
+ "epoch": 1.14,
618
+ "step": 65
619
+ },
620
+ {
621
+ "loss": 0.177,
622
+ "learning_rate": 7.2e-05,
623
+ "epoch": 1.16,
624
+ "step": 66
625
+ },
626
+ {
627
+ "loss": 0.2578,
628
+ "learning_rate": 7.2e-05,
629
+ "epoch": 1.18,
630
+ "step": 67
631
+ },
632
+ {
633
+ "loss": 0.3072,
634
+ "learning_rate": 7.2e-05,
635
+ "epoch": 1.19,
636
+ "step": 68
637
+ },
638
+ {
639
+ "loss": 0.1981,
640
+ "learning_rate": 7.2e-05,
641
+ "epoch": 1.21,
642
+ "step": 69
643
+ },
644
+ {
645
+ "loss": 0.0553,
646
+ "learning_rate": 7.2e-05,
647
+ "epoch": 1.23,
648
+ "step": 70
649
+ },
650
+ {
651
+ "loss": 0.1716,
652
+ "learning_rate": 7.2e-05,
653
+ "epoch": 1.25,
654
+ "step": 71
655
+ },
656
+ {
657
+ "loss": 0.2619,
658
+ "learning_rate": 7.2e-05,
659
+ "epoch": 1.26,
660
+ "step": 72
661
+ },
662
+ {
663
+ "loss": 0.2038,
664
+ "learning_rate": 7.2e-05,
665
+ "epoch": 1.28,
666
+ "step": 73
667
+ },
668
+ {
669
+ "loss": 0.28,
670
+ "learning_rate": 7.2e-05,
671
+ "epoch": 1.3,
672
+ "step": 74
673
+ },
674
+ {
675
+ "loss": 0.2674,
676
+ "learning_rate": 7.2e-05,
677
+ "epoch": 1.32,
678
+ "step": 75
679
+ },
680
+ {
681
+ "eval_word_swap_loss": 0.3006734848022461,
682
+ "eval_word_swap_score": -0.09580742567777634,
683
+ "eval_word_swap_brier_score": 0.09580742567777634,
684
+ "eval_word_swap_average_probability": 0.7959051728248596,
685
+ "eval_word_swap_accuracy": 0.91,
686
+ "eval_word_swap_probabilities": [
687
+ 0.9969964027404785,
688
+ 0.9507887959480286,
689
+ 0.9633437991142273,
690
+ 0.999962329864502,
691
+ 0.9999476671218872,
692
+ 0.9999223947525024,
693
+ 0.6215924024581909,
694
+ 0.3536596894264221,
695
+ 0.13465213775634766,
696
+ 0.7330600023269653,
697
+ 0.6277885437011719,
698
+ 0.8404280543327332,
699
+ 0.5336952209472656,
700
+ 0.5391131043434143,
701
+ 0.5149630904197693,
702
+ 0.9999996423721313,
703
+ 0.9999943971633911,
704
+ 0.9999983310699463,
705
+ 0.8160877823829651,
706
+ 0.9007056951522827,
707
+ 0.7626283168792725,
708
+ 0.9999878406524658,
709
+ 0.9999994039535522,
710
+ 0.9998726844787598,
711
+ 0.5319790244102478,
712
+ 0.6001496911048889,
713
+ 0.642182469367981,
714
+ 0.5197558999061584,
715
+ 0.5434504151344299,
716
+ 0.4450205862522125,
717
+ 0.8029510378837585,
718
+ 0.6582866311073303,
719
+ 0.8840171694755554,
720
+ 0.5005488395690918,
721
+ 0.40833598375320435,
722
+ 0.9256088733673096,
723
+ 0.49089136719703674,
724
+ 0.5098429322242737,
725
+ 0.5044549703598022,
726
+ 0.5252902507781982,
727
+ 0.15801160037517548,
728
+ 0.6188400983810425,
729
+ 0.9657522439956665,
730
+ 0.9947081804275513,
731
+ 0.8111419081687927,
732
+ 0.998430073261261,
733
+ 0.9975191354751587,
734
+ 0.9995943903923035,
735
+ 0.7320981025695801,
736
+ 0.9001358151435852,
737
+ 0.9640558958053589,
738
+ 0.7182196974754333,
739
+ 0.9855635166168213,
740
+ 0.2862946093082428,
741
+ 0.96307772397995,
742
+ 0.9980369210243225,
743
+ 0.9860389232635498,
744
+ 0.9999997615814209,
745
+ 0.9999990463256836,
746
+ 0.9999996423721313,
747
+ 0.7825421094894409,
748
+ 0.6848147511482239,
749
+ 0.9333890080451965,
750
+ 0.9774789214134216,
751
+ 0.9175354242324829,
752
+ 0.16475406289100647,
753
+ 0.9846394062042236,
754
+ 0.9522899389266968,
755
+ 0.9891321063041687,
756
+ 0.9987329840660095,
757
+ 0.9948404431343079,
758
+ 0.9994403719902039,
759
+ 0.8653655648231506,
760
+ 0.8490062355995178,
761
+ 0.8827810287475586,
762
+ 0.8907150030136108,
763
+ 0.6887461543083191,
764
+ 0.787915050983429,
765
+ 0.5203461050987244,
766
+ 0.5583745837211609,
767
+ 0.5360216498374939,
768
+ 0.9968532919883728,
769
+ 0.836186408996582,
770
+ 0.9695208072662354,
771
+ 0.9980870485305786,
772
+ 0.9990785121917725,
773
+ 0.9999843835830688,
774
+ 0.9948882460594177,
775
+ 0.9927409291267395,
776
+ 0.9828306436538696,
777
+ 0.512332558631897,
778
+ 0.5076123476028442,
779
+ 0.4851413667201996,
780
+ 0.9944328665733337,
781
+ 0.9997733235359192,
782
+ 0.9990137815475464,
783
+ 0.9985355138778687,
784
+ 0.9987162351608276,
785
+ 0.9776900410652161,
786
+ 0.5287637710571289
787
+ ],
788
+ "eval_word_swap_runtime": 10.8825,
789
+ "eval_word_swap_samples_per_second": 9.189,
790
+ "eval_word_swap_steps_per_second": 0.184,
791
+ "epoch": 1.32,
792
+ "step": 75
793
+ },
794
+ {
795
+ "loss": 0.1221,
796
+ "learning_rate": 7.2e-05,
797
+ "epoch": 1.33,
798
+ "step": 76
799
+ },
800
+ {
801
+ "loss": 0.0759,
802
+ "learning_rate": 7.2e-05,
803
+ "epoch": 1.35,
804
+ "step": 77
805
+ },
806
+ {
807
+ "loss": 0.1313,
808
+ "learning_rate": 7.2e-05,
809
+ "epoch": 1.37,
810
+ "step": 78
811
+ },
812
+ {
813
+ "loss": 0.2506,
814
+ "learning_rate": 7.2e-05,
815
+ "epoch": 1.39,
816
+ "step": 79
817
+ },
818
+ {
819
+ "loss": 0.3464,
820
+ "learning_rate": 7.2e-05,
821
+ "epoch": 1.4,
822
+ "step": 80
823
+ },
824
+ {
825
+ "loss": 0.2204,
826
+ "learning_rate": 7.2e-05,
827
+ "epoch": 1.42,
828
+ "step": 81
829
+ },
830
+ {
831
+ "loss": 0.1565,
832
+ "learning_rate": 7.2e-05,
833
+ "epoch": 1.44,
834
+ "step": 82
835
+ },
836
+ {
837
+ "loss": 0.218,
838
+ "learning_rate": 7.2e-05,
839
+ "epoch": 1.46,
840
+ "step": 83
841
+ },
842
+ {
843
+ "loss": 0.2444,
844
+ "learning_rate": 7.2e-05,
845
+ "epoch": 1.47,
846
+ "step": 84
847
+ },
848
+ {
849
+ "loss": 0.0554,
850
+ "learning_rate": 7.2e-05,
851
+ "epoch": 1.49,
852
+ "step": 85
853
+ },
854
+ {
855
+ "loss": 0.1428,
856
+ "learning_rate": 7.2e-05,
857
+ "epoch": 1.51,
858
+ "step": 86
859
+ },
860
+ {
861
+ "loss": 0.1364,
862
+ "learning_rate": 7.2e-05,
863
+ "epoch": 1.53,
864
+ "step": 87
865
+ },
866
+ {
867
+ "loss": 0.1239,
868
+ "learning_rate": 7.2e-05,
869
+ "epoch": 1.54,
870
+ "step": 88
871
+ },
872
+ {
873
+ "loss": 0.0936,
874
+ "learning_rate": 7.2e-05,
875
+ "epoch": 1.56,
876
+ "step": 89
877
+ },
878
+ {
879
+ "loss": 0.1616,
880
+ "learning_rate": 7.2e-05,
881
+ "epoch": 1.58,
882
+ "step": 90
883
+ },
884
+ {
885
+ "loss": 0.1755,
886
+ "learning_rate": 7.2e-05,
887
+ "epoch": 1.6,
888
+ "step": 91
889
+ },
890
+ {
891
+ "loss": 0.1327,
892
+ "learning_rate": 7.2e-05,
893
+ "epoch": 1.61,
894
+ "step": 92
895
+ },
896
+ {
897
+ "loss": 0.2136,
898
+ "learning_rate": 7.2e-05,
899
+ "epoch": 1.63,
900
+ "step": 93
901
+ },
902
+ {
903
+ "loss": 0.3009,
904
+ "learning_rate": 7.2e-05,
905
+ "epoch": 1.65,
906
+ "step": 94
907
+ },
908
+ {
909
+ "loss": 0.2399,
910
+ "learning_rate": 7.2e-05,
911
+ "epoch": 1.67,
912
+ "step": 95
913
+ },
914
+ {
915
+ "loss": 0.1839,
916
+ "learning_rate": 7.2e-05,
917
+ "epoch": 1.68,
918
+ "step": 96
919
+ },
920
+ {
921
+ "loss": 0.2827,
922
+ "learning_rate": 7.2e-05,
923
+ "epoch": 1.7,
924
+ "step": 97
925
+ },
926
+ {
927
+ "loss": 0.1555,
928
+ "learning_rate": 7.2e-05,
929
+ "epoch": 1.72,
930
+ "step": 98
931
+ },
932
+ {
933
+ "loss": 0.1261,
934
+ "learning_rate": 7.2e-05,
935
+ "epoch": 1.74,
936
+ "step": 99
937
+ },
938
+ {
939
+ "loss": 0.1755,
940
+ "learning_rate": 7.2e-05,
941
+ "epoch": 1.75,
942
+ "step": 100
943
+ },
944
+ {
945
+ "eval_word_swap_loss": 0.27881884574890137,
946
+ "eval_word_swap_score": -0.0890786200761795,
947
+ "eval_word_swap_brier_score": 0.0890786200761795,
948
+ "eval_word_swap_average_probability": 0.812042236328125,
949
+ "eval_word_swap_accuracy": 0.86,
950
+ "eval_word_swap_probabilities": [
951
+ 0.9990302324295044,
952
+ 0.9934167861938477,
953
+ 0.9882057905197144,
954
+ 0.9999992847442627,
955
+ 0.9999972581863403,
956
+ 0.9999982118606567,
957
+ 0.6618084907531738,
958
+ 0.46874576807022095,
959
+ 0.18735335767269135,
960
+ 0.7869884967803955,
961
+ 0.6931788325309753,
962
+ 0.8977562189102173,
963
+ 0.8259074091911316,
964
+ 0.6889633536338806,
965
+ 0.4997744560241699,
966
+ 0.9999847412109375,
967
+ 0.9997929930686951,
968
+ 0.9999535083770752,
969
+ 0.6179655194282532,
970
+ 0.9170421361923218,
971
+ 0.6882592439651489,
972
+ 0.9999262094497681,
973
+ 0.9999960660934448,
974
+ 0.9997277855873108,
975
+ 0.60245680809021,
976
+ 0.5533709526062012,
977
+ 0.5738738775253296,
978
+ 0.622164249420166,
979
+ 0.6978135704994202,
980
+ 0.5412458181381226,
981
+ 0.8736642003059387,
982
+ 0.6749231815338135,
983
+ 0.9533592462539673,
984
+ 0.4013397991657257,
985
+ 0.3414597511291504,
986
+ 0.9767069816589355,
987
+ 0.4634705185890198,
988
+ 0.45713555812835693,
989
+ 0.4639211893081665,
990
+ 0.5916439890861511,
991
+ 0.0867951288819313,
992
+ 0.6237714886665344,
993
+ 0.994411289691925,
994
+ 0.9996441602706909,
995
+ 0.9434242844581604,
996
+ 0.9911892414093018,
997
+ 0.9899294972419739,
998
+ 0.997734785079956,
999
+ 0.6482833027839661,
1000
+ 0.7949731945991516,
1001
+ 0.9145122766494751,
1002
+ 0.9854534268379211,
1003
+ 0.9998905658721924,
1004
+ 0.42005982995033264,
1005
+ 0.9783969521522522,
1006
+ 0.9939537644386292,
1007
+ 0.9943590760231018,
1008
+ 0.9999988079071045,
1009
+ 0.9999896287918091,
1010
+ 0.9999988079071045,
1011
+ 0.763102650642395,
1012
+ 0.6968801021575928,
1013
+ 0.9561358094215393,
1014
+ 0.9580368399620056,
1015
+ 0.8346080780029297,
1016
+ 0.18302346765995026,
1017
+ 0.9889246821403503,
1018
+ 0.9487248659133911,
1019
+ 0.985115110874176,
1020
+ 0.9980303645133972,
1021
+ 0.9961317777633667,
1022
+ 0.9992265701293945,
1023
+ 0.9698296785354614,
1024
+ 0.9555807709693909,
1025
+ 0.9703328609466553,
1026
+ 0.9328600764274597,
1027
+ 0.8323138356208801,
1028
+ 0.8644497394561768,
1029
+ 0.5253221988677979,
1030
+ 0.5801857709884644,
1031
+ 0.5422365069389343,
1032
+ 0.9996942281723022,
1033
+ 0.7747805118560791,
1034
+ 0.9802049994468689,
1035
+ 0.9969304203987122,
1036
+ 0.9993155002593994,
1037
+ 0.9998737573623657,
1038
+ 0.9990155696868896,
1039
+ 0.9991414546966553,
1040
+ 0.9988875985145569,
1041
+ 0.4802470803260803,
1042
+ 0.4588475823402405,
1043
+ 0.4658763110637665,
1044
+ 0.9996687173843384,
1045
+ 0.9999492168426514,
1046
+ 0.9997994303703308,
1047
+ 0.9986470341682434,
1048
+ 0.9975645542144775,
1049
+ 0.9981847405433655,
1050
+ 0.5174503922462463
1051
+ ],
1052
+ "eval_word_swap_runtime": 10.8837,
1053
+ "eval_word_swap_samples_per_second": 9.188,
1054
+ "eval_word_swap_steps_per_second": 0.184,
1055
+ "epoch": 1.75,
1056
+ "step": 100
1057
+ },
1058
+ {
1059
+ "train_runtime": 830.5373,
1060
+ "train_samples_per_second": 3.853,
1061
+ "train_steps_per_second": 0.12,
1062
+ "total_flos": 0.0,
1063
+ "train_loss": 0.391106785312295,
1064
+ "epoch": 1.75,
1065
+ "step": 100
1066
+ }
1067
+ ]