oza75 commited on
Commit
6f3d52e
·
verified ·
1 Parent(s): fe39d91

Training in progress, step 50000, checkpoint

Browse files
checkpoint-50000/added_tokens.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "<extra_id_0>": 259,
3
+ "<extra_id_100>": 359,
4
+ "<extra_id_101>": 360,
5
+ "<extra_id_102>": 361,
6
+ "<extra_id_103>": 362,
7
+ "<extra_id_104>": 363,
8
+ "<extra_id_105>": 364,
9
+ "<extra_id_106>": 365,
10
+ "<extra_id_107>": 366,
11
+ "<extra_id_108>": 367,
12
+ "<extra_id_109>": 368,
13
+ "<extra_id_10>": 269,
14
+ "<extra_id_110>": 369,
15
+ "<extra_id_111>": 370,
16
+ "<extra_id_112>": 371,
17
+ "<extra_id_113>": 372,
18
+ "<extra_id_114>": 373,
19
+ "<extra_id_115>": 374,
20
+ "<extra_id_116>": 375,
21
+ "<extra_id_117>": 376,
22
+ "<extra_id_118>": 377,
23
+ "<extra_id_119>": 378,
24
+ "<extra_id_11>": 270,
25
+ "<extra_id_120>": 379,
26
+ "<extra_id_121>": 380,
27
+ "<extra_id_122>": 381,
28
+ "<extra_id_123>": 382,
29
+ "<extra_id_124>": 383,
30
+ "<extra_id_12>": 271,
31
+ "<extra_id_13>": 272,
32
+ "<extra_id_14>": 273,
33
+ "<extra_id_15>": 274,
34
+ "<extra_id_16>": 275,
35
+ "<extra_id_17>": 276,
36
+ "<extra_id_18>": 277,
37
+ "<extra_id_19>": 278,
38
+ "<extra_id_1>": 260,
39
+ "<extra_id_20>": 279,
40
+ "<extra_id_21>": 280,
41
+ "<extra_id_22>": 281,
42
+ "<extra_id_23>": 282,
43
+ "<extra_id_24>": 283,
44
+ "<extra_id_25>": 284,
45
+ "<extra_id_26>": 285,
46
+ "<extra_id_27>": 286,
47
+ "<extra_id_28>": 287,
48
+ "<extra_id_29>": 288,
49
+ "<extra_id_2>": 261,
50
+ "<extra_id_30>": 289,
51
+ "<extra_id_31>": 290,
52
+ "<extra_id_32>": 291,
53
+ "<extra_id_33>": 292,
54
+ "<extra_id_34>": 293,
55
+ "<extra_id_35>": 294,
56
+ "<extra_id_36>": 295,
57
+ "<extra_id_37>": 296,
58
+ "<extra_id_38>": 297,
59
+ "<extra_id_39>": 298,
60
+ "<extra_id_3>": 262,
61
+ "<extra_id_40>": 299,
62
+ "<extra_id_41>": 300,
63
+ "<extra_id_42>": 301,
64
+ "<extra_id_43>": 302,
65
+ "<extra_id_44>": 303,
66
+ "<extra_id_45>": 304,
67
+ "<extra_id_46>": 305,
68
+ "<extra_id_47>": 306,
69
+ "<extra_id_48>": 307,
70
+ "<extra_id_49>": 308,
71
+ "<extra_id_4>": 263,
72
+ "<extra_id_50>": 309,
73
+ "<extra_id_51>": 310,
74
+ "<extra_id_52>": 311,
75
+ "<extra_id_53>": 312,
76
+ "<extra_id_54>": 313,
77
+ "<extra_id_55>": 314,
78
+ "<extra_id_56>": 315,
79
+ "<extra_id_57>": 316,
80
+ "<extra_id_58>": 317,
81
+ "<extra_id_59>": 318,
82
+ "<extra_id_5>": 264,
83
+ "<extra_id_60>": 319,
84
+ "<extra_id_61>": 320,
85
+ "<extra_id_62>": 321,
86
+ "<extra_id_63>": 322,
87
+ "<extra_id_64>": 323,
88
+ "<extra_id_65>": 324,
89
+ "<extra_id_66>": 325,
90
+ "<extra_id_67>": 326,
91
+ "<extra_id_68>": 327,
92
+ "<extra_id_69>": 328,
93
+ "<extra_id_6>": 265,
94
+ "<extra_id_70>": 329,
95
+ "<extra_id_71>": 330,
96
+ "<extra_id_72>": 331,
97
+ "<extra_id_73>": 332,
98
+ "<extra_id_74>": 333,
99
+ "<extra_id_75>": 334,
100
+ "<extra_id_76>": 335,
101
+ "<extra_id_77>": 336,
102
+ "<extra_id_78>": 337,
103
+ "<extra_id_79>": 338,
104
+ "<extra_id_7>": 266,
105
+ "<extra_id_80>": 339,
106
+ "<extra_id_81>": 340,
107
+ "<extra_id_82>": 341,
108
+ "<extra_id_83>": 342,
109
+ "<extra_id_84>": 343,
110
+ "<extra_id_85>": 344,
111
+ "<extra_id_86>": 345,
112
+ "<extra_id_87>": 346,
113
+ "<extra_id_88>": 347,
114
+ "<extra_id_89>": 348,
115
+ "<extra_id_8>": 267,
116
+ "<extra_id_90>": 349,
117
+ "<extra_id_91>": 350,
118
+ "<extra_id_92>": 351,
119
+ "<extra_id_93>": 352,
120
+ "<extra_id_94>": 353,
121
+ "<extra_id_95>": 354,
122
+ "<extra_id_96>": 355,
123
+ "<extra_id_97>": 356,
124
+ "<extra_id_98>": 357,
125
+ "<extra_id_99>": 358,
126
+ "<extra_id_9>": 268
127
+ }
checkpoint-50000/config.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-50000/configuration_sanigec.py ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Any
2
+
3
+ from transformers import PreTrainedConfig
4
+
5
+
6
+ class SaniGecConfig(PreTrainedConfig):
7
+ """
8
+ Configuration for SaniGec, the non-autoregressive byte-encoder GEC tagger.
9
+
10
+ A pretrained ByT5 encoder over raw UTF-8 bytes is pooled back to codepoint resolution and feeds
11
+ two heads: a binary error-detection head and an edit-correction head over the tag vocabulary
12
+ (KEEP/DELETE/APPEND_c/REPLACE_c). `backbone_name` is the HF checkpoint the encoder is first
13
+ transferred from (via `SaniGecModel.from_backbone`); `encoder_config` is that encoder's
14
+ serialized architecture, persisted so a reload rebuilds the encoder to the right shape offline
15
+ and the saved state-dict fills it — the model never re-fetches the backbone on reload.
16
+ `hidden_size` mirrors the encoder's `d_model` so the heads and the pooling anchor are sized to
17
+ it. The loss hyperparameters live here so a checkpoint records the objective it trained under.
18
+ """
19
+
20
+ model_type = "sani_gec"
21
+
22
+ def __init__(
23
+ self,
24
+ backbone_name: str = "google/byt5-base",
25
+ encoder_config: dict[str, Any] | None = None,
26
+ hidden_size: int = 1536,
27
+ num_correction_tags: int = 512,
28
+ correction_class_weights: list[float] | None = None,
29
+ dropout: float = 0.1,
30
+ detection_loss_weight: float = 1.0,
31
+ correction_loss_weight: float = 1.0,
32
+ focal_gamma: float = 2.0,
33
+ label_smoothing: float = 0.1,
34
+ keep_confidence_margin: float = 0.0,
35
+ max_correction_passes: int = 12,
36
+ pad_token_id: int = 0,
37
+ **kwargs: Any,
38
+ ) -> None:
39
+ self.backbone_name = backbone_name
40
+ self.encoder_config = encoder_config
41
+ # The encoder's d_model is authoritative when its architecture is known; the default is only
42
+ # a placeholder for a bare config built before a backbone is attached.
43
+ self.hidden_size = encoder_config["d_model"] if encoder_config else hidden_size
44
+ self.num_correction_tags = num_correction_tags
45
+ self.correction_class_weights = correction_class_weights
46
+ self.dropout = dropout
47
+ self.detection_loss_weight = detection_loss_weight
48
+ self.correction_loss_weight = correction_loss_weight
49
+ self.focal_gamma = focal_gamma
50
+ self.label_smoothing = label_smoothing
51
+ # Inference-time decode rule (the iterative corrector reads these), persisted so a reloaded
52
+ # checkpoint corrects with no external arguments.
53
+ self.keep_confidence_margin = keep_confidence_margin
54
+ self.max_correction_passes = max_correction_passes
55
+ # Forward via a dict so the special tokens ride in **kwargs: the transformers v5 type stub
56
+ # does not name pad_token_id on PreTrainedConfig.__init__, though the runtime accepts it.
57
+ init_kwargs: dict[str, Any] = {"pad_token_id": pad_token_id, **kwargs}
58
+ super().__init__(**init_kwargs)
checkpoint-50000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4e8aa83c3bdf31f37cf2575cdbc16e4162f85051ae1ed26ab229f563c3d20d2
3
+ size 1674814416
checkpoint-50000/modeling_sanigec.py ADDED
@@ -0,0 +1,302 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from dataclasses import dataclass
3
+ from pathlib import Path
4
+ from typing import Any
5
+
6
+ import torch
7
+ from torch import nn
8
+ from transformers import (
9
+ AutoConfig,
10
+ AutoModel,
11
+ PreTrainedModel,
12
+ PreTrainedTokenizerBase,
13
+ T5Config,
14
+ T5EncoderModel,
15
+ )
16
+ from transformers.modeling_outputs import ModelOutput # type: ignore[attr-defined]
17
+ from transformers.pipelines import PIPELINE_REGISTRY
18
+ from transformers.utils import cached_file, logging # type: ignore[attr-defined]
19
+
20
+ from bambara_gec.model.byte_bridge import ByteCodepointBridge
21
+ from bambara_gec.model.configuration_sanigec import SaniGecConfig
22
+ from bambara_gec.model.losses import (
23
+ IGNORE_INDEX,
24
+ ClassWeightedLabelSmoothedCrossEntropy,
25
+ FocalLoss,
26
+ combine_head_losses,
27
+ )
28
+ from bambara_gec.model.pipeline import TASK_NAME, SaniGecCorrectionPipeline
29
+ from bambara_gec.model.tags import TagScheme
30
+ from bambara_gec.model.vocabulary import CharVocabulary
31
+
32
+ VOCAB_FILE = "vocab.json"
33
+
34
+ # Shipped in config.json so `pipeline("sani-gec", model=repo, trust_remote_code=True)` resolves the
35
+ # correction pipeline from the repo's own pipeline.py — the same self-containment contract as
36
+ # auto_map for the config/model.
37
+ CUSTOM_PIPELINES = {
38
+ TASK_NAME: {
39
+ "impl": "pipeline.SaniGecCorrectionPipeline",
40
+ "pt": ["AutoModel"],
41
+ "tf": [],
42
+ }
43
+ }
44
+
45
+ logger = logging.get_logger(__name__)
46
+
47
+
48
+ @dataclass
49
+ class SaniGecOutput(ModelOutput):
50
+ """
51
+ Forward output: the combined training loss (None without labels) and the two heads' per-position
52
+ logits — detection over {keep, edit} and correction over the tag vocabulary — at codepoint
53
+ resolution (the pooled `[anchor, cp₁…cpₙ, eos]` layout).
54
+ """
55
+
56
+ loss: torch.Tensor | None = None
57
+ detection_logits: torch.Tensor | None = None
58
+ correction_logits: torch.Tensor | None = None
59
+
60
+
61
+ class SaniGecModel(PreTrainedModel):
62
+ """
63
+ A pretrained ByT5 encoder over raw UTF-8 bytes, pooled back to codepoint resolution by the
64
+ byte→codepoint bridge, feeding the two GECToR-style heads. Non-autoregressive: every codepoint
65
+ is tagged in parallel in one forward; multi-pass correction is the inference loop's job.
66
+
67
+ The encoder is built empty from the persisted `encoder_config` here and the pretrained weights
68
+ are transferred once by `from_backbone` (not inside `__init__`): so reloading a fine-tuned
69
+ checkpoint rebuilds the encoder to shape and lets the saved state-dict fill it, instead of
70
+ re-fetching the backbone and overwriting the trained weights. Only the two heads are
71
+ `_init_weights`-initialised (the framework hook); the encoder owns its own init/loaded weights.
72
+ The detection/correction labels are codepoint-indexed and align 1:1 with the pooled layout, so
73
+ the loss reshapes over codepoint positions exactly as before.
74
+ """
75
+
76
+ config_class = SaniGecConfig
77
+ base_model_prefix = "sani_gec"
78
+
79
+ def __init__(self, config: SaniGecConfig) -> None:
80
+ super().__init__(config)
81
+ encoder_config = (
82
+ T5Config.from_dict(config.encoder_config)
83
+ if config.encoder_config is not None
84
+ else T5Config.from_pretrained(config.backbone_name)
85
+ )
86
+ self.encoder = T5EncoderModel(encoder_config)
87
+ hidden = encoder_config.d_model
88
+ self.bridge = ByteCodepointBridge(hidden)
89
+ self.dropout = nn.Dropout(config.dropout)
90
+ self.detection_head = nn.Linear(hidden, 2)
91
+ self.correction_head = nn.Linear(hidden, config.num_correction_tags)
92
+ self.detection_loss = FocalLoss(gamma=config.focal_gamma)
93
+ correction_class_weights = (
94
+ torch.tensor(config.correction_class_weights, dtype=torch.float32)
95
+ if config.correction_class_weights is not None
96
+ else None
97
+ )
98
+ self.correction_loss = ClassWeightedLabelSmoothedCrossEntropy(
99
+ class_weight=correction_class_weights, smoothing=config.label_smoothing
100
+ )
101
+ # The decode tag scheme (id -> applicable edit). None until attached from a CharVocabulary
102
+ # (at construction) or rebuilt from the shipped vocab.json (on reload).
103
+ self.tag_scheme: TagScheme | None = None
104
+ # The byte tokenizer (stock ByT5). Carried so save_pretrained ships it beside the weights,
105
+ # making the repo loadable end-to-end by `pipeline(...)` / AutoTokenizer with no external
106
+ # reference.
107
+ self.tokenizer: PreTrainedTokenizerBase | None = None
108
+ self.post_init()
109
+
110
+ @classmethod
111
+ def from_backbone(
112
+ cls,
113
+ config: SaniGecConfig,
114
+ tag_scheme: TagScheme | None = None,
115
+ tokenizer: PreTrainedTokenizerBase | None = None,
116
+ ) -> "SaniGecModel":
117
+ """
118
+ First construction: load the pretrained ByT5 encoder from `config.backbone_name` exactly
119
+ once, record its architecture into the config (so later reloads rebuild it offline), build
120
+ the model with a matching empty encoder, and transfer the pretrained weights into it. A tag
121
+ scheme and/or tokenizer given here are attached so the saved model is self-contained.
122
+ """
123
+ # Force safetensors: transformers refuses torch.load of a .bin checkpoint under torch < 2.6
124
+ # (CVE-2025-32434), and the byt5 checkpoints ship safetensors — which also skips the
125
+ # redundant multi-GB .bin download.
126
+ pretrained = T5EncoderModel.from_pretrained(config.backbone_name, use_safetensors=True)
127
+ config.encoder_config = pretrained.config.to_dict()
128
+ config.hidden_size = pretrained.config.d_model
129
+ config.custom_pipelines = CUSTOM_PIPELINES
130
+ model = cls(config)
131
+ model.encoder.load_state_dict(pretrained.state_dict())
132
+ if tag_scheme is not None:
133
+ model.set_tag_scheme(tag_scheme)
134
+ if tokenizer is not None:
135
+ model.tokenizer = tokenizer
136
+ return model
137
+
138
+ def set_tag_scheme(self, tag_scheme: TagScheme) -> None:
139
+ """
140
+ Attach the decode scheme and derive the config's label maps from it. The correction head is
141
+ already sized to `num_correction_tags`, so the scheme's tag count must match it — a mismatch
142
+ would mean the head and the decode vocabulary disagree, which is asserted rather than
143
+ silently shipped.
144
+ """
145
+ if tag_scheme.num_tags != self.config.num_correction_tags:
146
+ raise ValueError(
147
+ f"tag scheme has {tag_scheme.num_tags} tags but the correction head is sized for "
148
+ f"{self.config.num_correction_tags}"
149
+ )
150
+ self.tag_scheme = tag_scheme
151
+ self.config.id2label = tag_scheme.id2label
152
+ self.config.label2id = tag_scheme.label2id
153
+
154
+ def save_pretrained( # type: ignore[override]
155
+ self, save_directory: str | os.PathLike[str], **kwargs: Any
156
+ ) -> None:
157
+ """
158
+ Ship the tag vocabulary as vocab.json beside the weights and config, so a reload rebuilds
159
+ the identical decode scheme from the repo alone (the self-containment contract). It is
160
+ written before the weights/config so it is on disk for any subsequent folder upload; the
161
+ Trainer's `push_to_hub` (US-09-5) uploads the whole folder, so the pushed repo carries it.
162
+ """
163
+ if self.tag_scheme is not None:
164
+ Path(save_directory).mkdir(parents=True, exist_ok=True)
165
+ self.tag_scheme.vocabulary.save(Path(save_directory) / VOCAB_FILE)
166
+ if self.tokenizer is not None:
167
+ self.tokenizer.save_pretrained(str(save_directory))
168
+ super().save_pretrained(save_directory, **kwargs)
169
+
170
+ @classmethod
171
+ def from_pretrained( # type: ignore[override]
172
+ cls, pretrained_model_name_or_path: str, *model_args: Any, **kwargs: Any
173
+ ) -> "SaniGecModel":
174
+ """
175
+ Load the model and, when the repo ships a vocab.json, rebuild and attach the decode scheme
176
+ from it (re-deriving the config label maps), so inference needs no external vocabulary. A
177
+ missing vocab.json is warned about rather than failed: the model still does forward passes
178
+ but cannot decode until a scheme is attached. Does not support `output_loading_info=True`.
179
+ """
180
+ model = super().from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
181
+ vocab_path = cached_file(
182
+ pretrained_model_name_or_path,
183
+ VOCAB_FILE,
184
+ _raise_exceptions_for_missing_entries=False,
185
+ _raise_exceptions_for_connection_errors=False,
186
+ )
187
+ if vocab_path is not None:
188
+ model.set_tag_scheme(TagScheme(CharVocabulary.load(Path(vocab_path))))
189
+ else:
190
+ logger.warning(
191
+ "no %s alongside %s; model loaded without a decode tag scheme (cannot correct "
192
+ "until one is attached)",
193
+ VOCAB_FILE,
194
+ pretrained_model_name_or_path,
195
+ )
196
+ return model
197
+
198
+ def _init_weights(self, module: nn.Module) -> None:
199
+ """
200
+ The PreTrainedModel init hook, narrowed to exactly the two task heads (identity, not type):
201
+ post_init() runs this over every submodule, but the encoder owns its own T5 init (then
202
+ loaded weights), so only the fresh heads are (re)initialised — the encoder's structure-aware
203
+ init is never overwritten by a generic normal().
204
+ """
205
+ if module is self.detection_head or module is self.correction_head:
206
+ assert isinstance(module, nn.Linear)
207
+ module.weight.data.normal_(mean=0.0, std=0.02)
208
+ if module.bias is not None:
209
+ module.bias.data.zero_()
210
+
211
+ def gradient_checkpointing_enable(
212
+ self, gradient_checkpointing_kwargs: dict[str, object] | None = None
213
+ ) -> None:
214
+ """
215
+ Trade encoder activation memory for recomputation. Only the deep ByT5 encoder benefits (its
216
+ long byte sequences dominate activation memory); the two small heads are not checkpointed,
217
+ so this delegates straight to the encoder.
218
+ """
219
+ self.encoder.gradient_checkpointing_enable(
220
+ gradient_checkpointing_kwargs=gradient_checkpointing_kwargs
221
+ )
222
+
223
+ def forward(
224
+ self,
225
+ input_ids: torch.Tensor,
226
+ attention_mask: torch.Tensor | None = None,
227
+ codepoint_segments: torch.Tensor | None = None,
228
+ detection_labels: torch.Tensor | None = None,
229
+ correction_labels: torch.Tensor | None = None,
230
+ instance_weight: torch.Tensor | None = None,
231
+ max_codepoints: int | None = None,
232
+ ) -> SaniGecOutput:
233
+ if codepoint_segments is None:
234
+ raise ValueError("codepoint_segments is required to pool byte hidden-states")
235
+ byte_hidden = self.encoder(
236
+ input_ids=input_ids, attention_mask=attention_mask
237
+ ).last_hidden_state
238
+ if max_codepoints is None:
239
+ # Prefer the label width (host-side, no device sync); fall back to the segment max only
240
+ # when there are no labels and the caller did not pass the width explicitly.
241
+ max_codepoints = (
242
+ detection_labels.shape[1]
243
+ if detection_labels is not None
244
+ else int(codepoint_segments.max().item()) + 1
245
+ )
246
+ pooled = self.dropout(
247
+ self.bridge.pool_batch(byte_hidden, codepoint_segments, max_codepoints)
248
+ )
249
+ detection_logits = self.detection_head(pooled)
250
+ correction_logits = self.correction_head(pooled)
251
+
252
+ loss = None
253
+ if detection_labels is not None and correction_labels is not None:
254
+ batch, length, _ = pooled.shape
255
+ weights = None
256
+ if instance_weight is not None:
257
+ weights = instance_weight.unsqueeze(1).expand(batch, length).reshape(-1)
258
+ detection_loss = self.detection_loss(
259
+ detection_logits.reshape(-1, 2), detection_labels.reshape(-1), weights
260
+ )
261
+ correction_loss = self.correction_loss(
262
+ correction_logits.reshape(-1, self.config.num_correction_tags),
263
+ correction_labels.reshape(-1),
264
+ weights,
265
+ )
266
+ loss = combine_head_losses(
267
+ detection_loss,
268
+ correction_loss,
269
+ self.config.detection_loss_weight,
270
+ self.config.correction_loss_weight,
271
+ )
272
+
273
+ return SaniGecOutput(
274
+ loss=loss,
275
+ detection_logits=detection_logits,
276
+ correction_logits=correction_logits,
277
+ )
278
+
279
+
280
+ SaniGecConfig.register_for_auto_class()
281
+ SaniGecModel.register_for_auto_class("AutoModel")
282
+ try:
283
+ AutoConfig.register("sani_gec", SaniGecConfig)
284
+ AutoModel.register(SaniGecConfig, SaniGecModel)
285
+ except ValueError:
286
+ # Already registered (the module was imported more than once in this process).
287
+ pass
288
+
289
+ # Register the correction pipeline here, not in pipeline.py, so it can bind to SaniGecModel without
290
+ # pipeline.py importing this module (which would be a cycle). Importing the model is enough to make
291
+ # `pipeline("sani-gec", model=…)` resolvable in-process. This in-process registration and the
292
+ # shipped CUSTOM_PIPELINES config block above are two faces of the same wiring (task "sani-gec",
293
+ # AutoModel-backed text pipeline) and must be kept in sync by hand.
294
+ PIPELINE_REGISTRY.register_pipeline(
295
+ TASK_NAME,
296
+ pipeline_class=SaniGecCorrectionPipeline,
297
+ pt_model=SaniGecModel,
298
+ type="text",
299
+ )
300
+
301
+
302
+ __all__ = ["IGNORE_INDEX", "SaniGecConfig", "SaniGecModel", "SaniGecOutput"]
checkpoint-50000/tokenizer_config.json ADDED
@@ -0,0 +1,1162 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<pad>",
5
+ "lstrip": true,
6
+ "normalized": true,
7
+ "rstrip": true,
8
+ "single_word": false,
9
+ "special": false
10
+ },
11
+ "1": {
12
+ "content": "</s>",
13
+ "lstrip": true,
14
+ "normalized": true,
15
+ "rstrip": true,
16
+ "single_word": false,
17
+ "special": false
18
+ },
19
+ "2": {
20
+ "content": "<unk>",
21
+ "lstrip": true,
22
+ "normalized": true,
23
+ "rstrip": true,
24
+ "single_word": false,
25
+ "special": false
26
+ },
27
+ "259": {
28
+ "content": "<extra_id_0>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "260": {
36
+ "content": "<extra_id_1>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "261": {
44
+ "content": "<extra_id_2>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "262": {
52
+ "content": "<extra_id_3>",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "263": {
60
+ "content": "<extra_id_4>",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ },
67
+ "264": {
68
+ "content": "<extra_id_5>",
69
+ "lstrip": false,
70
+ "normalized": false,
71
+ "rstrip": false,
72
+ "single_word": false,
73
+ "special": true
74
+ },
75
+ "265": {
76
+ "content": "<extra_id_6>",
77
+ "lstrip": false,
78
+ "normalized": false,
79
+ "rstrip": false,
80
+ "single_word": false,
81
+ "special": true
82
+ },
83
+ "266": {
84
+ "content": "<extra_id_7>",
85
+ "lstrip": false,
86
+ "normalized": false,
87
+ "rstrip": false,
88
+ "single_word": false,
89
+ "special": true
90
+ },
91
+ "267": {
92
+ "content": "<extra_id_8>",
93
+ "lstrip": false,
94
+ "normalized": false,
95
+ "rstrip": false,
96
+ "single_word": false,
97
+ "special": true
98
+ },
99
+ "268": {
100
+ "content": "<extra_id_9>",
101
+ "lstrip": false,
102
+ "normalized": false,
103
+ "rstrip": false,
104
+ "single_word": false,
105
+ "special": true
106
+ },
107
+ "269": {
108
+ "content": "<extra_id_10>",
109
+ "lstrip": false,
110
+ "normalized": false,
111
+ "rstrip": false,
112
+ "single_word": false,
113
+ "special": true
114
+ },
115
+ "270": {
116
+ "content": "<extra_id_11>",
117
+ "lstrip": false,
118
+ "normalized": false,
119
+ "rstrip": false,
120
+ "single_word": false,
121
+ "special": true
122
+ },
123
+ "271": {
124
+ "content": "<extra_id_12>",
125
+ "lstrip": false,
126
+ "normalized": false,
127
+ "rstrip": false,
128
+ "single_word": false,
129
+ "special": true
130
+ },
131
+ "272": {
132
+ "content": "<extra_id_13>",
133
+ "lstrip": false,
134
+ "normalized": false,
135
+ "rstrip": false,
136
+ "single_word": false,
137
+ "special": true
138
+ },
139
+ "273": {
140
+ "content": "<extra_id_14>",
141
+ "lstrip": false,
142
+ "normalized": false,
143
+ "rstrip": false,
144
+ "single_word": false,
145
+ "special": true
146
+ },
147
+ "274": {
148
+ "content": "<extra_id_15>",
149
+ "lstrip": false,
150
+ "normalized": false,
151
+ "rstrip": false,
152
+ "single_word": false,
153
+ "special": true
154
+ },
155
+ "275": {
156
+ "content": "<extra_id_16>",
157
+ "lstrip": false,
158
+ "normalized": false,
159
+ "rstrip": false,
160
+ "single_word": false,
161
+ "special": true
162
+ },
163
+ "276": {
164
+ "content": "<extra_id_17>",
165
+ "lstrip": false,
166
+ "normalized": false,
167
+ "rstrip": false,
168
+ "single_word": false,
169
+ "special": true
170
+ },
171
+ "277": {
172
+ "content": "<extra_id_18>",
173
+ "lstrip": false,
174
+ "normalized": false,
175
+ "rstrip": false,
176
+ "single_word": false,
177
+ "special": true
178
+ },
179
+ "278": {
180
+ "content": "<extra_id_19>",
181
+ "lstrip": false,
182
+ "normalized": false,
183
+ "rstrip": false,
184
+ "single_word": false,
185
+ "special": true
186
+ },
187
+ "279": {
188
+ "content": "<extra_id_20>",
189
+ "lstrip": false,
190
+ "normalized": false,
191
+ "rstrip": false,
192
+ "single_word": false,
193
+ "special": true
194
+ },
195
+ "280": {
196
+ "content": "<extra_id_21>",
197
+ "lstrip": false,
198
+ "normalized": false,
199
+ "rstrip": false,
200
+ "single_word": false,
201
+ "special": true
202
+ },
203
+ "281": {
204
+ "content": "<extra_id_22>",
205
+ "lstrip": false,
206
+ "normalized": false,
207
+ "rstrip": false,
208
+ "single_word": false,
209
+ "special": true
210
+ },
211
+ "282": {
212
+ "content": "<extra_id_23>",
213
+ "lstrip": false,
214
+ "normalized": false,
215
+ "rstrip": false,
216
+ "single_word": false,
217
+ "special": true
218
+ },
219
+ "283": {
220
+ "content": "<extra_id_24>",
221
+ "lstrip": false,
222
+ "normalized": false,
223
+ "rstrip": false,
224
+ "single_word": false,
225
+ "special": true
226
+ },
227
+ "284": {
228
+ "content": "<extra_id_25>",
229
+ "lstrip": false,
230
+ "normalized": false,
231
+ "rstrip": false,
232
+ "single_word": false,
233
+ "special": true
234
+ },
235
+ "285": {
236
+ "content": "<extra_id_26>",
237
+ "lstrip": false,
238
+ "normalized": false,
239
+ "rstrip": false,
240
+ "single_word": false,
241
+ "special": true
242
+ },
243
+ "286": {
244
+ "content": "<extra_id_27>",
245
+ "lstrip": false,
246
+ "normalized": false,
247
+ "rstrip": false,
248
+ "single_word": false,
249
+ "special": true
250
+ },
251
+ "287": {
252
+ "content": "<extra_id_28>",
253
+ "lstrip": false,
254
+ "normalized": false,
255
+ "rstrip": false,
256
+ "single_word": false,
257
+ "special": true
258
+ },
259
+ "288": {
260
+ "content": "<extra_id_29>",
261
+ "lstrip": false,
262
+ "normalized": false,
263
+ "rstrip": false,
264
+ "single_word": false,
265
+ "special": true
266
+ },
267
+ "289": {
268
+ "content": "<extra_id_30>",
269
+ "lstrip": false,
270
+ "normalized": false,
271
+ "rstrip": false,
272
+ "single_word": false,
273
+ "special": true
274
+ },
275
+ "290": {
276
+ "content": "<extra_id_31>",
277
+ "lstrip": false,
278
+ "normalized": false,
279
+ "rstrip": false,
280
+ "single_word": false,
281
+ "special": true
282
+ },
283
+ "291": {
284
+ "content": "<extra_id_32>",
285
+ "lstrip": false,
286
+ "normalized": false,
287
+ "rstrip": false,
288
+ "single_word": false,
289
+ "special": true
290
+ },
291
+ "292": {
292
+ "content": "<extra_id_33>",
293
+ "lstrip": false,
294
+ "normalized": false,
295
+ "rstrip": false,
296
+ "single_word": false,
297
+ "special": true
298
+ },
299
+ "293": {
300
+ "content": "<extra_id_34>",
301
+ "lstrip": false,
302
+ "normalized": false,
303
+ "rstrip": false,
304
+ "single_word": false,
305
+ "special": true
306
+ },
307
+ "294": {
308
+ "content": "<extra_id_35>",
309
+ "lstrip": false,
310
+ "normalized": false,
311
+ "rstrip": false,
312
+ "single_word": false,
313
+ "special": true
314
+ },
315
+ "295": {
316
+ "content": "<extra_id_36>",
317
+ "lstrip": false,
318
+ "normalized": false,
319
+ "rstrip": false,
320
+ "single_word": false,
321
+ "special": true
322
+ },
323
+ "296": {
324
+ "content": "<extra_id_37>",
325
+ "lstrip": false,
326
+ "normalized": false,
327
+ "rstrip": false,
328
+ "single_word": false,
329
+ "special": true
330
+ },
331
+ "297": {
332
+ "content": "<extra_id_38>",
333
+ "lstrip": false,
334
+ "normalized": false,
335
+ "rstrip": false,
336
+ "single_word": false,
337
+ "special": true
338
+ },
339
+ "298": {
340
+ "content": "<extra_id_39>",
341
+ "lstrip": false,
342
+ "normalized": false,
343
+ "rstrip": false,
344
+ "single_word": false,
345
+ "special": true
346
+ },
347
+ "299": {
348
+ "content": "<extra_id_40>",
349
+ "lstrip": false,
350
+ "normalized": false,
351
+ "rstrip": false,
352
+ "single_word": false,
353
+ "special": true
354
+ },
355
+ "300": {
356
+ "content": "<extra_id_41>",
357
+ "lstrip": false,
358
+ "normalized": false,
359
+ "rstrip": false,
360
+ "single_word": false,
361
+ "special": true
362
+ },
363
+ "301": {
364
+ "content": "<extra_id_42>",
365
+ "lstrip": false,
366
+ "normalized": false,
367
+ "rstrip": false,
368
+ "single_word": false,
369
+ "special": true
370
+ },
371
+ "302": {
372
+ "content": "<extra_id_43>",
373
+ "lstrip": false,
374
+ "normalized": false,
375
+ "rstrip": false,
376
+ "single_word": false,
377
+ "special": true
378
+ },
379
+ "303": {
380
+ "content": "<extra_id_44>",
381
+ "lstrip": false,
382
+ "normalized": false,
383
+ "rstrip": false,
384
+ "single_word": false,
385
+ "special": true
386
+ },
387
+ "304": {
388
+ "content": "<extra_id_45>",
389
+ "lstrip": false,
390
+ "normalized": false,
391
+ "rstrip": false,
392
+ "single_word": false,
393
+ "special": true
394
+ },
395
+ "305": {
396
+ "content": "<extra_id_46>",
397
+ "lstrip": false,
398
+ "normalized": false,
399
+ "rstrip": false,
400
+ "single_word": false,
401
+ "special": true
402
+ },
403
+ "306": {
404
+ "content": "<extra_id_47>",
405
+ "lstrip": false,
406
+ "normalized": false,
407
+ "rstrip": false,
408
+ "single_word": false,
409
+ "special": true
410
+ },
411
+ "307": {
412
+ "content": "<extra_id_48>",
413
+ "lstrip": false,
414
+ "normalized": false,
415
+ "rstrip": false,
416
+ "single_word": false,
417
+ "special": true
418
+ },
419
+ "308": {
420
+ "content": "<extra_id_49>",
421
+ "lstrip": false,
422
+ "normalized": false,
423
+ "rstrip": false,
424
+ "single_word": false,
425
+ "special": true
426
+ },
427
+ "309": {
428
+ "content": "<extra_id_50>",
429
+ "lstrip": false,
430
+ "normalized": false,
431
+ "rstrip": false,
432
+ "single_word": false,
433
+ "special": true
434
+ },
435
+ "310": {
436
+ "content": "<extra_id_51>",
437
+ "lstrip": false,
438
+ "normalized": false,
439
+ "rstrip": false,
440
+ "single_word": false,
441
+ "special": true
442
+ },
443
+ "311": {
444
+ "content": "<extra_id_52>",
445
+ "lstrip": false,
446
+ "normalized": false,
447
+ "rstrip": false,
448
+ "single_word": false,
449
+ "special": true
450
+ },
451
+ "312": {
452
+ "content": "<extra_id_53>",
453
+ "lstrip": false,
454
+ "normalized": false,
455
+ "rstrip": false,
456
+ "single_word": false,
457
+ "special": true
458
+ },
459
+ "313": {
460
+ "content": "<extra_id_54>",
461
+ "lstrip": false,
462
+ "normalized": false,
463
+ "rstrip": false,
464
+ "single_word": false,
465
+ "special": true
466
+ },
467
+ "314": {
468
+ "content": "<extra_id_55>",
469
+ "lstrip": false,
470
+ "normalized": false,
471
+ "rstrip": false,
472
+ "single_word": false,
473
+ "special": true
474
+ },
475
+ "315": {
476
+ "content": "<extra_id_56>",
477
+ "lstrip": false,
478
+ "normalized": false,
479
+ "rstrip": false,
480
+ "single_word": false,
481
+ "special": true
482
+ },
483
+ "316": {
484
+ "content": "<extra_id_57>",
485
+ "lstrip": false,
486
+ "normalized": false,
487
+ "rstrip": false,
488
+ "single_word": false,
489
+ "special": true
490
+ },
491
+ "317": {
492
+ "content": "<extra_id_58>",
493
+ "lstrip": false,
494
+ "normalized": false,
495
+ "rstrip": false,
496
+ "single_word": false,
497
+ "special": true
498
+ },
499
+ "318": {
500
+ "content": "<extra_id_59>",
501
+ "lstrip": false,
502
+ "normalized": false,
503
+ "rstrip": false,
504
+ "single_word": false,
505
+ "special": true
506
+ },
507
+ "319": {
508
+ "content": "<extra_id_60>",
509
+ "lstrip": false,
510
+ "normalized": false,
511
+ "rstrip": false,
512
+ "single_word": false,
513
+ "special": true
514
+ },
515
+ "320": {
516
+ "content": "<extra_id_61>",
517
+ "lstrip": false,
518
+ "normalized": false,
519
+ "rstrip": false,
520
+ "single_word": false,
521
+ "special": true
522
+ },
523
+ "321": {
524
+ "content": "<extra_id_62>",
525
+ "lstrip": false,
526
+ "normalized": false,
527
+ "rstrip": false,
528
+ "single_word": false,
529
+ "special": true
530
+ },
531
+ "322": {
532
+ "content": "<extra_id_63>",
533
+ "lstrip": false,
534
+ "normalized": false,
535
+ "rstrip": false,
536
+ "single_word": false,
537
+ "special": true
538
+ },
539
+ "323": {
540
+ "content": "<extra_id_64>",
541
+ "lstrip": false,
542
+ "normalized": false,
543
+ "rstrip": false,
544
+ "single_word": false,
545
+ "special": true
546
+ },
547
+ "324": {
548
+ "content": "<extra_id_65>",
549
+ "lstrip": false,
550
+ "normalized": false,
551
+ "rstrip": false,
552
+ "single_word": false,
553
+ "special": true
554
+ },
555
+ "325": {
556
+ "content": "<extra_id_66>",
557
+ "lstrip": false,
558
+ "normalized": false,
559
+ "rstrip": false,
560
+ "single_word": false,
561
+ "special": true
562
+ },
563
+ "326": {
564
+ "content": "<extra_id_67>",
565
+ "lstrip": false,
566
+ "normalized": false,
567
+ "rstrip": false,
568
+ "single_word": false,
569
+ "special": true
570
+ },
571
+ "327": {
572
+ "content": "<extra_id_68>",
573
+ "lstrip": false,
574
+ "normalized": false,
575
+ "rstrip": false,
576
+ "single_word": false,
577
+ "special": true
578
+ },
579
+ "328": {
580
+ "content": "<extra_id_69>",
581
+ "lstrip": false,
582
+ "normalized": false,
583
+ "rstrip": false,
584
+ "single_word": false,
585
+ "special": true
586
+ },
587
+ "329": {
588
+ "content": "<extra_id_70>",
589
+ "lstrip": false,
590
+ "normalized": false,
591
+ "rstrip": false,
592
+ "single_word": false,
593
+ "special": true
594
+ },
595
+ "330": {
596
+ "content": "<extra_id_71>",
597
+ "lstrip": false,
598
+ "normalized": false,
599
+ "rstrip": false,
600
+ "single_word": false,
601
+ "special": true
602
+ },
603
+ "331": {
604
+ "content": "<extra_id_72>",
605
+ "lstrip": false,
606
+ "normalized": false,
607
+ "rstrip": false,
608
+ "single_word": false,
609
+ "special": true
610
+ },
611
+ "332": {
612
+ "content": "<extra_id_73>",
613
+ "lstrip": false,
614
+ "normalized": false,
615
+ "rstrip": false,
616
+ "single_word": false,
617
+ "special": true
618
+ },
619
+ "333": {
620
+ "content": "<extra_id_74>",
621
+ "lstrip": false,
622
+ "normalized": false,
623
+ "rstrip": false,
624
+ "single_word": false,
625
+ "special": true
626
+ },
627
+ "334": {
628
+ "content": "<extra_id_75>",
629
+ "lstrip": false,
630
+ "normalized": false,
631
+ "rstrip": false,
632
+ "single_word": false,
633
+ "special": true
634
+ },
635
+ "335": {
636
+ "content": "<extra_id_76>",
637
+ "lstrip": false,
638
+ "normalized": false,
639
+ "rstrip": false,
640
+ "single_word": false,
641
+ "special": true
642
+ },
643
+ "336": {
644
+ "content": "<extra_id_77>",
645
+ "lstrip": false,
646
+ "normalized": false,
647
+ "rstrip": false,
648
+ "single_word": false,
649
+ "special": true
650
+ },
651
+ "337": {
652
+ "content": "<extra_id_78>",
653
+ "lstrip": false,
654
+ "normalized": false,
655
+ "rstrip": false,
656
+ "single_word": false,
657
+ "special": true
658
+ },
659
+ "338": {
660
+ "content": "<extra_id_79>",
661
+ "lstrip": false,
662
+ "normalized": false,
663
+ "rstrip": false,
664
+ "single_word": false,
665
+ "special": true
666
+ },
667
+ "339": {
668
+ "content": "<extra_id_80>",
669
+ "lstrip": false,
670
+ "normalized": false,
671
+ "rstrip": false,
672
+ "single_word": false,
673
+ "special": true
674
+ },
675
+ "340": {
676
+ "content": "<extra_id_81>",
677
+ "lstrip": false,
678
+ "normalized": false,
679
+ "rstrip": false,
680
+ "single_word": false,
681
+ "special": true
682
+ },
683
+ "341": {
684
+ "content": "<extra_id_82>",
685
+ "lstrip": false,
686
+ "normalized": false,
687
+ "rstrip": false,
688
+ "single_word": false,
689
+ "special": true
690
+ },
691
+ "342": {
692
+ "content": "<extra_id_83>",
693
+ "lstrip": false,
694
+ "normalized": false,
695
+ "rstrip": false,
696
+ "single_word": false,
697
+ "special": true
698
+ },
699
+ "343": {
700
+ "content": "<extra_id_84>",
701
+ "lstrip": false,
702
+ "normalized": false,
703
+ "rstrip": false,
704
+ "single_word": false,
705
+ "special": true
706
+ },
707
+ "344": {
708
+ "content": "<extra_id_85>",
709
+ "lstrip": false,
710
+ "normalized": false,
711
+ "rstrip": false,
712
+ "single_word": false,
713
+ "special": true
714
+ },
715
+ "345": {
716
+ "content": "<extra_id_86>",
717
+ "lstrip": false,
718
+ "normalized": false,
719
+ "rstrip": false,
720
+ "single_word": false,
721
+ "special": true
722
+ },
723
+ "346": {
724
+ "content": "<extra_id_87>",
725
+ "lstrip": false,
726
+ "normalized": false,
727
+ "rstrip": false,
728
+ "single_word": false,
729
+ "special": true
730
+ },
731
+ "347": {
732
+ "content": "<extra_id_88>",
733
+ "lstrip": false,
734
+ "normalized": false,
735
+ "rstrip": false,
736
+ "single_word": false,
737
+ "special": true
738
+ },
739
+ "348": {
740
+ "content": "<extra_id_89>",
741
+ "lstrip": false,
742
+ "normalized": false,
743
+ "rstrip": false,
744
+ "single_word": false,
745
+ "special": true
746
+ },
747
+ "349": {
748
+ "content": "<extra_id_90>",
749
+ "lstrip": false,
750
+ "normalized": false,
751
+ "rstrip": false,
752
+ "single_word": false,
753
+ "special": true
754
+ },
755
+ "350": {
756
+ "content": "<extra_id_91>",
757
+ "lstrip": false,
758
+ "normalized": false,
759
+ "rstrip": false,
760
+ "single_word": false,
761
+ "special": true
762
+ },
763
+ "351": {
764
+ "content": "<extra_id_92>",
765
+ "lstrip": false,
766
+ "normalized": false,
767
+ "rstrip": false,
768
+ "single_word": false,
769
+ "special": true
770
+ },
771
+ "352": {
772
+ "content": "<extra_id_93>",
773
+ "lstrip": false,
774
+ "normalized": false,
775
+ "rstrip": false,
776
+ "single_word": false,
777
+ "special": true
778
+ },
779
+ "353": {
780
+ "content": "<extra_id_94>",
781
+ "lstrip": false,
782
+ "normalized": false,
783
+ "rstrip": false,
784
+ "single_word": false,
785
+ "special": true
786
+ },
787
+ "354": {
788
+ "content": "<extra_id_95>",
789
+ "lstrip": false,
790
+ "normalized": false,
791
+ "rstrip": false,
792
+ "single_word": false,
793
+ "special": true
794
+ },
795
+ "355": {
796
+ "content": "<extra_id_96>",
797
+ "lstrip": false,
798
+ "normalized": false,
799
+ "rstrip": false,
800
+ "single_word": false,
801
+ "special": true
802
+ },
803
+ "356": {
804
+ "content": "<extra_id_97>",
805
+ "lstrip": false,
806
+ "normalized": false,
807
+ "rstrip": false,
808
+ "single_word": false,
809
+ "special": true
810
+ },
811
+ "357": {
812
+ "content": "<extra_id_98>",
813
+ "lstrip": false,
814
+ "normalized": false,
815
+ "rstrip": false,
816
+ "single_word": false,
817
+ "special": true
818
+ },
819
+ "358": {
820
+ "content": "<extra_id_99>",
821
+ "lstrip": false,
822
+ "normalized": false,
823
+ "rstrip": false,
824
+ "single_word": false,
825
+ "special": true
826
+ },
827
+ "359": {
828
+ "content": "<extra_id_100>",
829
+ "lstrip": false,
830
+ "normalized": false,
831
+ "rstrip": false,
832
+ "single_word": false,
833
+ "special": true
834
+ },
835
+ "360": {
836
+ "content": "<extra_id_101>",
837
+ "lstrip": false,
838
+ "normalized": false,
839
+ "rstrip": false,
840
+ "single_word": false,
841
+ "special": true
842
+ },
843
+ "361": {
844
+ "content": "<extra_id_102>",
845
+ "lstrip": false,
846
+ "normalized": false,
847
+ "rstrip": false,
848
+ "single_word": false,
849
+ "special": true
850
+ },
851
+ "362": {
852
+ "content": "<extra_id_103>",
853
+ "lstrip": false,
854
+ "normalized": false,
855
+ "rstrip": false,
856
+ "single_word": false,
857
+ "special": true
858
+ },
859
+ "363": {
860
+ "content": "<extra_id_104>",
861
+ "lstrip": false,
862
+ "normalized": false,
863
+ "rstrip": false,
864
+ "single_word": false,
865
+ "special": true
866
+ },
867
+ "364": {
868
+ "content": "<extra_id_105>",
869
+ "lstrip": false,
870
+ "normalized": false,
871
+ "rstrip": false,
872
+ "single_word": false,
873
+ "special": true
874
+ },
875
+ "365": {
876
+ "content": "<extra_id_106>",
877
+ "lstrip": false,
878
+ "normalized": false,
879
+ "rstrip": false,
880
+ "single_word": false,
881
+ "special": true
882
+ },
883
+ "366": {
884
+ "content": "<extra_id_107>",
885
+ "lstrip": false,
886
+ "normalized": false,
887
+ "rstrip": false,
888
+ "single_word": false,
889
+ "special": true
890
+ },
891
+ "367": {
892
+ "content": "<extra_id_108>",
893
+ "lstrip": false,
894
+ "normalized": false,
895
+ "rstrip": false,
896
+ "single_word": false,
897
+ "special": true
898
+ },
899
+ "368": {
900
+ "content": "<extra_id_109>",
901
+ "lstrip": false,
902
+ "normalized": false,
903
+ "rstrip": false,
904
+ "single_word": false,
905
+ "special": true
906
+ },
907
+ "369": {
908
+ "content": "<extra_id_110>",
909
+ "lstrip": false,
910
+ "normalized": false,
911
+ "rstrip": false,
912
+ "single_word": false,
913
+ "special": true
914
+ },
915
+ "370": {
916
+ "content": "<extra_id_111>",
917
+ "lstrip": false,
918
+ "normalized": false,
919
+ "rstrip": false,
920
+ "single_word": false,
921
+ "special": true
922
+ },
923
+ "371": {
924
+ "content": "<extra_id_112>",
925
+ "lstrip": false,
926
+ "normalized": false,
927
+ "rstrip": false,
928
+ "single_word": false,
929
+ "special": true
930
+ },
931
+ "372": {
932
+ "content": "<extra_id_113>",
933
+ "lstrip": false,
934
+ "normalized": false,
935
+ "rstrip": false,
936
+ "single_word": false,
937
+ "special": true
938
+ },
939
+ "373": {
940
+ "content": "<extra_id_114>",
941
+ "lstrip": false,
942
+ "normalized": false,
943
+ "rstrip": false,
944
+ "single_word": false,
945
+ "special": true
946
+ },
947
+ "374": {
948
+ "content": "<extra_id_115>",
949
+ "lstrip": false,
950
+ "normalized": false,
951
+ "rstrip": false,
952
+ "single_word": false,
953
+ "special": true
954
+ },
955
+ "375": {
956
+ "content": "<extra_id_116>",
957
+ "lstrip": false,
958
+ "normalized": false,
959
+ "rstrip": false,
960
+ "single_word": false,
961
+ "special": true
962
+ },
963
+ "376": {
964
+ "content": "<extra_id_117>",
965
+ "lstrip": false,
966
+ "normalized": false,
967
+ "rstrip": false,
968
+ "single_word": false,
969
+ "special": true
970
+ },
971
+ "377": {
972
+ "content": "<extra_id_118>",
973
+ "lstrip": false,
974
+ "normalized": false,
975
+ "rstrip": false,
976
+ "single_word": false,
977
+ "special": true
978
+ },
979
+ "378": {
980
+ "content": "<extra_id_119>",
981
+ "lstrip": false,
982
+ "normalized": false,
983
+ "rstrip": false,
984
+ "single_word": false,
985
+ "special": true
986
+ },
987
+ "379": {
988
+ "content": "<extra_id_120>",
989
+ "lstrip": false,
990
+ "normalized": false,
991
+ "rstrip": false,
992
+ "single_word": false,
993
+ "special": true
994
+ },
995
+ "380": {
996
+ "content": "<extra_id_121>",
997
+ "lstrip": false,
998
+ "normalized": false,
999
+ "rstrip": false,
1000
+ "single_word": false,
1001
+ "special": true
1002
+ },
1003
+ "381": {
1004
+ "content": "<extra_id_122>",
1005
+ "lstrip": false,
1006
+ "normalized": false,
1007
+ "rstrip": false,
1008
+ "single_word": false,
1009
+ "special": true
1010
+ },
1011
+ "382": {
1012
+ "content": "<extra_id_123>",
1013
+ "lstrip": false,
1014
+ "normalized": false,
1015
+ "rstrip": false,
1016
+ "single_word": false,
1017
+ "special": true
1018
+ },
1019
+ "383": {
1020
+ "content": "<extra_id_124>",
1021
+ "lstrip": false,
1022
+ "normalized": false,
1023
+ "rstrip": false,
1024
+ "single_word": false,
1025
+ "special": true
1026
+ }
1027
+ },
1028
+ "backend": "custom",
1029
+ "eos_token": "</s>",
1030
+ "extra_ids": 0,
1031
+ "extra_special_tokens": [
1032
+ "<extra_id_0>",
1033
+ "<extra_id_1>",
1034
+ "<extra_id_2>",
1035
+ "<extra_id_3>",
1036
+ "<extra_id_4>",
1037
+ "<extra_id_5>",
1038
+ "<extra_id_6>",
1039
+ "<extra_id_7>",
1040
+ "<extra_id_8>",
1041
+ "<extra_id_9>",
1042
+ "<extra_id_10>",
1043
+ "<extra_id_11>",
1044
+ "<extra_id_12>",
1045
+ "<extra_id_13>",
1046
+ "<extra_id_14>",
1047
+ "<extra_id_15>",
1048
+ "<extra_id_16>",
1049
+ "<extra_id_17>",
1050
+ "<extra_id_18>",
1051
+ "<extra_id_19>",
1052
+ "<extra_id_20>",
1053
+ "<extra_id_21>",
1054
+ "<extra_id_22>",
1055
+ "<extra_id_23>",
1056
+ "<extra_id_24>",
1057
+ "<extra_id_25>",
1058
+ "<extra_id_26>",
1059
+ "<extra_id_27>",
1060
+ "<extra_id_28>",
1061
+ "<extra_id_29>",
1062
+ "<extra_id_30>",
1063
+ "<extra_id_31>",
1064
+ "<extra_id_32>",
1065
+ "<extra_id_33>",
1066
+ "<extra_id_34>",
1067
+ "<extra_id_35>",
1068
+ "<extra_id_36>",
1069
+ "<extra_id_37>",
1070
+ "<extra_id_38>",
1071
+ "<extra_id_39>",
1072
+ "<extra_id_40>",
1073
+ "<extra_id_41>",
1074
+ "<extra_id_42>",
1075
+ "<extra_id_43>",
1076
+ "<extra_id_44>",
1077
+ "<extra_id_45>",
1078
+ "<extra_id_46>",
1079
+ "<extra_id_47>",
1080
+ "<extra_id_48>",
1081
+ "<extra_id_49>",
1082
+ "<extra_id_50>",
1083
+ "<extra_id_51>",
1084
+ "<extra_id_52>",
1085
+ "<extra_id_53>",
1086
+ "<extra_id_54>",
1087
+ "<extra_id_55>",
1088
+ "<extra_id_56>",
1089
+ "<extra_id_57>",
1090
+ "<extra_id_58>",
1091
+ "<extra_id_59>",
1092
+ "<extra_id_60>",
1093
+ "<extra_id_61>",
1094
+ "<extra_id_62>",
1095
+ "<extra_id_63>",
1096
+ "<extra_id_64>",
1097
+ "<extra_id_65>",
1098
+ "<extra_id_66>",
1099
+ "<extra_id_67>",
1100
+ "<extra_id_68>",
1101
+ "<extra_id_69>",
1102
+ "<extra_id_70>",
1103
+ "<extra_id_71>",
1104
+ "<extra_id_72>",
1105
+ "<extra_id_73>",
1106
+ "<extra_id_74>",
1107
+ "<extra_id_75>",
1108
+ "<extra_id_76>",
1109
+ "<extra_id_77>",
1110
+ "<extra_id_78>",
1111
+ "<extra_id_79>",
1112
+ "<extra_id_80>",
1113
+ "<extra_id_81>",
1114
+ "<extra_id_82>",
1115
+ "<extra_id_83>",
1116
+ "<extra_id_84>",
1117
+ "<extra_id_85>",
1118
+ "<extra_id_86>",
1119
+ "<extra_id_87>",
1120
+ "<extra_id_88>",
1121
+ "<extra_id_89>",
1122
+ "<extra_id_90>",
1123
+ "<extra_id_91>",
1124
+ "<extra_id_92>",
1125
+ "<extra_id_93>",
1126
+ "<extra_id_94>",
1127
+ "<extra_id_95>",
1128
+ "<extra_id_96>",
1129
+ "<extra_id_97>",
1130
+ "<extra_id_98>",
1131
+ "<extra_id_99>",
1132
+ "<extra_id_100>",
1133
+ "<extra_id_101>",
1134
+ "<extra_id_102>",
1135
+ "<extra_id_103>",
1136
+ "<extra_id_104>",
1137
+ "<extra_id_105>",
1138
+ "<extra_id_106>",
1139
+ "<extra_id_107>",
1140
+ "<extra_id_108>",
1141
+ "<extra_id_109>",
1142
+ "<extra_id_110>",
1143
+ "<extra_id_111>",
1144
+ "<extra_id_112>",
1145
+ "<extra_id_113>",
1146
+ "<extra_id_114>",
1147
+ "<extra_id_115>",
1148
+ "<extra_id_116>",
1149
+ "<extra_id_117>",
1150
+ "<extra_id_118>",
1151
+ "<extra_id_119>",
1152
+ "<extra_id_120>",
1153
+ "<extra_id_121>",
1154
+ "<extra_id_122>",
1155
+ "<extra_id_123>",
1156
+ "<extra_id_124>"
1157
+ ],
1158
+ "model_max_length": 1000000000000000019884624838656,
1159
+ "pad_token": "<pad>",
1160
+ "tokenizer_class": "ByT5Tokenizer",
1161
+ "unk_token": "<unk>"
1162
+ }
checkpoint-50000/trainer_state.json ADDED
@@ -0,0 +1,1294 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.8981985915225603,
6
+ "eval_steps": 2500,
7
+ "global_step": 50000,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.01898217573698297,
14
+ "grad_norm": 2.8858866691589355,
15
+ "learning_rate": 4.735843087630496e-05,
16
+ "loss": 6.00573583984375,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 0.03796435147396594,
21
+ "grad_norm": 0.22312745451927185,
22
+ "learning_rate": 9.481176842771275e-05,
23
+ "loss": 1.7886568603515625,
24
+ "step": 1000
25
+ },
26
+ {
27
+ "epoch": 0.05694652721094892,
28
+ "grad_norm": 0.3883834183216095,
29
+ "learning_rate": 0.0001422651059791205,
30
+ "loss": 1.4103663330078124,
31
+ "step": 1500
32
+ },
33
+ {
34
+ "epoch": 0.07592870294793189,
35
+ "grad_norm": 0.28461548686027527,
36
+ "learning_rate": 0.0001897184435305283,
37
+ "loss": 1.2999722900390625,
38
+ "step": 2000
39
+ },
40
+ {
41
+ "epoch": 0.09491087868491487,
42
+ "grad_norm": 0.3026827871799469,
43
+ "learning_rate": 0.00023717178108193608,
44
+ "loss": 1.27368310546875,
45
+ "step": 2500
46
+ },
47
+ {
48
+ "epoch": 0.09491087868491487,
49
+ "eval_correction_accuracy": 0.9691768119898284,
50
+ "eval_detection_f1": 0.5029481803464545,
51
+ "eval_loss": 1.2979440689086914,
52
+ "eval_runtime": 60.3464,
53
+ "eval_samples_per_second": 371.886,
54
+ "eval_steps_per_second": 2.916,
55
+ "step": 2500
56
+ },
57
+ {
58
+ "epoch": 0.09491087868491487,
59
+ "eval_edit_f0_5": 0.8420576085111199,
60
+ "eval_edit_f0_5_m0": 0.8057950832842375,
61
+ "eval_edit_f0_5_m0.1": 0.8175929454030809,
62
+ "eval_edit_f0_5_m0.2": 0.8316255884796455,
63
+ "eval_edit_f0_5_m0.3": 0.8420576085111199,
64
+ "step": 2500
65
+ },
66
+ {
67
+ "epoch": 0.09491087868491487,
68
+ "eval_real_f0_5": 0.42910447761194037,
69
+ "eval_real_f0_5_m0": 0.3888400702987698,
70
+ "eval_real_f0_5_m0.1": 0.4028041825095057,
71
+ "eval_real_f0_5_m0.2": 0.4175050301810866,
72
+ "eval_real_f0_5_m0.3": 0.42910447761194037,
73
+ "step": 2500
74
+ },
75
+ {
76
+ "epoch": 0.11389305442189784,
77
+ "grad_norm": 0.27056726813316345,
78
+ "learning_rate": 0.00028462511863334386,
79
+ "loss": 1.2630726318359375,
80
+ "step": 3000
81
+ },
82
+ {
83
+ "epoch": 0.1328752301588808,
84
+ "grad_norm": 0.2008272260427475,
85
+ "learning_rate": 0.00029996551748314967,
86
+ "loss": 1.2586767578125,
87
+ "step": 3500
88
+ },
89
+ {
90
+ "epoch": 0.15185740589586377,
91
+ "grad_norm": 0.23608435690402985,
92
+ "learning_rate": 0.00029978808195120705,
93
+ "loss": 1.2485474853515626,
94
+ "step": 4000
95
+ },
96
+ {
97
+ "epoch": 0.17083958163284677,
98
+ "grad_norm": 0.17198926210403442,
99
+ "learning_rate": 0.00029945995039741313,
100
+ "loss": 1.2438245849609375,
101
+ "step": 4500
102
+ },
103
+ {
104
+ "epoch": 0.18982175736982973,
105
+ "grad_norm": 0.20551930367946625,
106
+ "learning_rate": 0.00029898145294228977,
107
+ "loss": 1.2398516845703125,
108
+ "step": 5000
109
+ },
110
+ {
111
+ "epoch": 0.18982175736982973,
112
+ "eval_correction_accuracy": 0.9716923312269569,
113
+ "eval_detection_f1": 0.5269904148108618,
114
+ "eval_loss": 1.2654461860656738,
115
+ "eval_runtime": 59.8792,
116
+ "eval_samples_per_second": 374.788,
117
+ "eval_steps_per_second": 2.939,
118
+ "step": 5000
119
+ },
120
+ {
121
+ "epoch": 0.18982175736982973,
122
+ "eval_edit_f0_5": 0.872664843158936,
123
+ "eval_edit_f0_5_m0": 0.8547840106743758,
124
+ "eval_edit_f0_5_m0.1": 0.8604271144418775,
125
+ "eval_edit_f0_5_m0.2": 0.8685822219678324,
126
+ "eval_edit_f0_5_m0.3": 0.872664843158936,
127
+ "step": 5000
128
+ },
129
+ {
130
+ "epoch": 0.18982175736982973,
131
+ "eval_real_f0_5": 0.46151439299123903,
132
+ "eval_real_f0_5_m0": 0.4191919191919192,
133
+ "eval_real_f0_5_m0.1": 0.4297297297297297,
134
+ "eval_real_f0_5_m0.2": 0.445906432748538,
135
+ "eval_real_f0_5_m0.3": 0.46151439299123903,
136
+ "step": 5000
137
+ },
138
+ {
139
+ "epoch": 0.2088039331068127,
140
+ "grad_norm": 0.20124033093452454,
141
+ "learning_rate": 0.00029835307098370265,
142
+ "loss": 1.2357362060546875,
143
+ "step": 5500
144
+ },
145
+ {
146
+ "epoch": 0.2277861088437957,
147
+ "grad_norm": 0.23115237057209015,
148
+ "learning_rate": 0.0002975754367125453,
149
+ "loss": 1.2375308837890624,
150
+ "step": 6000
151
+ },
152
+ {
153
+ "epoch": 0.24676828458077865,
154
+ "grad_norm": 0.1954943835735321,
155
+ "learning_rate": 0.00029664933247671615,
156
+ "loss": 1.2349964599609375,
157
+ "step": 6500
158
+ },
159
+ {
160
+ "epoch": 0.2657504603177616,
161
+ "grad_norm": 0.17400752007961273,
162
+ "learning_rate": 0.0002955756899940283,
163
+ "loss": 1.2303714599609374,
164
+ "step": 7000
165
+ },
166
+ {
167
+ "epoch": 0.2847326360547446,
168
+ "grad_norm": 0.17888489365577698,
169
+ "learning_rate": 0.00029435558941484435,
170
+ "loss": 1.2309805908203124,
171
+ "step": 7500
172
+ },
173
+ {
174
+ "epoch": 0.2847326360547446,
175
+ "eval_correction_accuracy": 0.9723985228660039,
176
+ "eval_detection_f1": 0.5397412768051171,
177
+ "eval_loss": 1.2546594142913818,
178
+ "eval_runtime": 60.3599,
179
+ "eval_samples_per_second": 371.803,
180
+ "eval_steps_per_second": 2.916,
181
+ "step": 7500
182
+ },
183
+ {
184
+ "epoch": 0.2847326360547446,
185
+ "eval_edit_f0_5": 0.8861485517636936,
186
+ "eval_edit_f0_5_m0": 0.8749653451621846,
187
+ "eval_edit_f0_5_m0.1": 0.8788065728226123,
188
+ "eval_edit_f0_5_m0.2": 0.8814529381764608,
189
+ "eval_edit_f0_5_m0.3": 0.8861485517636936,
190
+ "step": 7500
191
+ },
192
+ {
193
+ "epoch": 0.2847326360547446,
194
+ "eval_real_f0_5": 0.47371638141809297,
195
+ "eval_real_f0_5_m0": 0.4501055966209081,
196
+ "eval_real_f0_5_m0.1": 0.4613259668508288,
197
+ "eval_real_f0_5_m0.2": 0.47164351851851855,
198
+ "eval_real_f0_5_m0.3": 0.47371638141809297,
199
+ "step": 7500
200
+ },
201
+ {
202
+ "epoch": 0.30371481179172755,
203
+ "grad_norm": 0.1810256987810135,
204
+ "learning_rate": 0.0002929902582353787,
205
+ "loss": 1.227394775390625,
206
+ "step": 8000
207
+ },
208
+ {
209
+ "epoch": 0.32269698752871057,
210
+ "grad_norm": 0.18248461186885834,
211
+ "learning_rate": 0.00029148107006276056,
212
+ "loss": 1.22420361328125,
213
+ "step": 8500
214
+ },
215
+ {
216
+ "epoch": 0.34167916326569353,
217
+ "grad_norm": 0.1456211358308792,
218
+ "learning_rate": 0.0002898295432331011,
219
+ "loss": 1.22528125,
220
+ "step": 9000
221
+ },
222
+ {
223
+ "epoch": 0.3606613390026765,
224
+ "grad_norm": 0.1682305634021759,
225
+ "learning_rate": 0.0002880373392839537,
226
+ "loss": 1.2209486083984376,
227
+ "step": 9500
228
+ },
229
+ {
230
+ "epoch": 0.37964351473965946,
231
+ "grad_norm": 0.1455032080411911,
232
+ "learning_rate": 0.00028610626128270495,
233
+ "loss": 1.222905029296875,
234
+ "step": 10000
235
+ },
236
+ {
237
+ "epoch": 0.37964351473965946,
238
+ "eval_correction_accuracy": 0.9729365736386112,
239
+ "eval_detection_f1": 0.5878967328292231,
240
+ "eval_loss": 1.2507761716842651,
241
+ "eval_runtime": 60.7194,
242
+ "eval_samples_per_second": 369.602,
243
+ "eval_steps_per_second": 2.899,
244
+ "step": 10000
245
+ },
246
+ {
247
+ "epoch": 0.37964351473965946,
248
+ "eval_edit_f0_5": 0.8773855767606823,
249
+ "eval_edit_f0_5_m0": 0.8642143753752115,
250
+ "eval_edit_f0_5_m0.1": 0.871961638097338,
251
+ "eval_edit_f0_5_m0.2": 0.8739938939772413,
252
+ "eval_edit_f0_5_m0.3": 0.8773855767606823,
253
+ "step": 10000
254
+ },
255
+ {
256
+ "epoch": 0.37964351473965946,
257
+ "eval_real_f0_5": 0.43992027334851935,
258
+ "eval_real_f0_5_m0": 0.4214697406340058,
259
+ "eval_real_f0_5_m0.1": 0.42792792792792794,
260
+ "eval_real_f0_5_m0.2": 0.42728237791932056,
261
+ "eval_real_f0_5_m0.3": 0.43992027334851935,
262
+ "step": 10000
263
+ },
264
+ {
265
+ "epoch": 0.3986256904766424,
266
+ "grad_norm": 0.20864084362983704,
267
+ "learning_rate": 0.0002840382520125783,
268
+ "loss": 1.22280712890625,
269
+ "step": 10500
270
+ },
271
+ {
272
+ "epoch": 0.4176078662136254,
273
+ "grad_norm": 0.19003528356552124,
274
+ "learning_rate": 0.0002818353920180744,
275
+ "loss": 1.221659423828125,
276
+ "step": 11000
277
+ },
278
+ {
279
+ "epoch": 0.43659004195060835,
280
+ "grad_norm": 0.1750059872865677,
281
+ "learning_rate": 0.0002794998975118153,
282
+ "loss": 1.2191646728515626,
283
+ "step": 11500
284
+ },
285
+ {
286
+ "epoch": 0.4555722176875914,
287
+ "grad_norm": 0.13223567605018616,
288
+ "learning_rate": 0.00027703411814489855,
289
+ "loss": 1.216736328125,
290
+ "step": 12000
291
+ },
292
+ {
293
+ "epoch": 0.47455439342457434,
294
+ "grad_norm": 0.10733171552419662,
295
+ "learning_rate": 0.00027444053464300346,
296
+ "loss": 1.2142664794921876,
297
+ "step": 12500
298
+ },
299
+ {
300
+ "epoch": 0.47455439342457434,
301
+ "eval_correction_accuracy": 0.9733371445263059,
302
+ "eval_detection_f1": 0.5704304192239259,
303
+ "eval_loss": 1.2477121353149414,
304
+ "eval_runtime": 60.6105,
305
+ "eval_samples_per_second": 370.266,
306
+ "eval_steps_per_second": 2.904,
307
+ "step": 12500
308
+ },
309
+ {
310
+ "epoch": 0.47455439342457434,
311
+ "eval_edit_f0_5": 0.8926609218668673,
312
+ "eval_edit_f0_5_m0": 0.8845489496851842,
313
+ "eval_edit_f0_5_m0.1": 0.8883563574851988,
314
+ "eval_edit_f0_5_m0.2": 0.89086859688196,
315
+ "eval_edit_f0_5_m0.3": 0.8926609218668673,
316
+ "step": 12500
317
+ },
318
+ {
319
+ "epoch": 0.47455439342457434,
320
+ "eval_real_f0_5": 0.4787581699346405,
321
+ "eval_real_f0_5_m0": 0.43419633225458465,
322
+ "eval_real_f0_5_m0.1": 0.44776119402985065,
323
+ "eval_real_f0_5_m0.2": 0.46760391198044005,
324
+ "eval_real_f0_5_m0.3": 0.4787581699346405,
325
+ "step": 12500
326
+ },
327
+ {
328
+ "epoch": 0.4935365691615573,
329
+ "grad_norm": 0.1579447090625763,
330
+ "learning_rate": 0.0002717217563106287,
331
+ "loss": 1.21451904296875,
332
+ "step": 13000
333
+ },
334
+ {
335
+ "epoch": 0.5125187448985403,
336
+ "grad_norm": 0.13867036998271942,
337
+ "learning_rate": 0.00026888051840597135,
338
+ "loss": 1.2117508544921876,
339
+ "step": 13500
340
+ },
341
+ {
342
+ "epoch": 0.5315009206355232,
343
+ "grad_norm": 0.11431417614221573,
344
+ "learning_rate": 0.00026591967938908897,
345
+ "loss": 1.2146973876953124,
346
+ "step": 14000
347
+ },
348
+ {
349
+ "epoch": 0.5504830963725063,
350
+ "grad_norm": 0.14605163037776947,
351
+ "learning_rate": 0.00026284221804611323,
352
+ "loss": 1.211010498046875,
353
+ "step": 14500
354
+ },
355
+ {
356
+ "epoch": 0.5694652721094892,
357
+ "grad_norm": 0.13300460577011108,
358
+ "learning_rate": 0.0002596512304924075,
359
+ "loss": 1.2157716064453126,
360
+ "step": 15000
361
+ },
362
+ {
363
+ "epoch": 0.5694652721094892,
364
+ "eval_correction_accuracy": 0.9736186980555992,
365
+ "eval_detection_f1": 0.6009406304526895,
366
+ "eval_loss": 1.2431864738464355,
367
+ "eval_runtime": 60.2876,
368
+ "eval_samples_per_second": 372.249,
369
+ "eval_steps_per_second": 2.919,
370
+ "step": 15000
371
+ },
372
+ {
373
+ "epoch": 0.5694652721094892,
374
+ "eval_edit_f0_5": 0.8888329698596432,
375
+ "eval_edit_f0_5_m0": 0.8778231292517006,
376
+ "eval_edit_f0_5_m0.1": 0.8819654131210541,
377
+ "eval_edit_f0_5_m0.2": 0.8855925167432335,
378
+ "eval_edit_f0_5_m0.3": 0.8888329698596432,
379
+ "step": 15000
380
+ },
381
+ {
382
+ "epoch": 0.5694652721094892,
383
+ "eval_real_f0_5": 0.4537767756482525,
384
+ "eval_real_f0_5_m0": 0.4154693486590038,
385
+ "eval_real_f0_5_m0.1": 0.42251755265797397,
386
+ "eval_real_f0_5_m0.2": 0.4458512931034483,
387
+ "eval_real_f0_5_m0.3": 0.4537767756482525,
388
+ "step": 15000
389
+ },
390
+ {
391
+ "epoch": 0.5884474478464722,
392
+ "grad_norm": 0.16023032367229462,
393
+ "learning_rate": 0.0002563499270576846,
394
+ "loss": 1.21418017578125,
395
+ "step": 15500
396
+ },
397
+ {
398
+ "epoch": 0.6074296235834551,
399
+ "grad_norm": 0.14038224518299103,
400
+ "learning_rate": 0.0002529416290562178,
401
+ "loss": 1.2102242431640624,
402
+ "step": 16000
403
+ },
404
+ {
405
+ "epoch": 0.6264117993204381,
406
+ "grad_norm": 0.19150033593177795,
407
+ "learning_rate": 0.00024942976544539404,
408
+ "loss": 1.21311474609375,
409
+ "step": 16500
410
+ },
411
+ {
412
+ "epoch": 0.6453939750574211,
413
+ "grad_norm": 0.13594546914100647,
414
+ "learning_rate": 0.0002458178693759725,
415
+ "loss": 1.2101019287109376,
416
+ "step": 17000
417
+ },
418
+ {
419
+ "epoch": 0.664376150794404,
420
+ "grad_norm": 0.0971856564283371,
421
+ "learning_rate": 0.00024210957463751702,
422
+ "loss": 1.2085975341796875,
423
+ "step": 17500
424
+ },
425
+ {
426
+ "epoch": 0.664376150794404,
427
+ "eval_correction_accuracy": 0.9739441000853563,
428
+ "eval_detection_f1": 0.6041727242827385,
429
+ "eval_loss": 1.2378469705581665,
430
+ "eval_runtime": 60.7237,
431
+ "eval_samples_per_second": 369.575,
432
+ "eval_steps_per_second": 2.898,
433
+ "step": 17500
434
+ },
435
+ {
436
+ "epoch": 0.664376150794404,
437
+ "eval_edit_f0_5": 0.8950703830407717,
438
+ "eval_edit_f0_5_m0": 0.8823850308221047,
439
+ "eval_edit_f0_5_m0.1": 0.8891276795062545,
440
+ "eval_edit_f0_5_m0.2": 0.891679139641887,
441
+ "eval_edit_f0_5_m0.3": 0.8950703830407717,
442
+ "step": 17500
443
+ },
444
+ {
445
+ "epoch": 0.664376150794404,
446
+ "eval_real_f0_5": 0.45221843003412976,
447
+ "eval_real_f0_5_m0": 0.4207080504364695,
448
+ "eval_real_f0_5_m0.1": 0.4268916155419223,
449
+ "eval_real_f0_5_m0.2": 0.43951612903225806,
450
+ "eval_real_f0_5_m0.3": 0.45221843003412976,
451
+ "step": 17500
452
+ },
453
+ {
454
+ "epoch": 0.6833583265313871,
455
+ "grad_norm": 0.23436909914016724,
456
+ "learning_rate": 0.00023830861200258084,
457
+ "loss": 1.2103984375,
458
+ "step": 18000
459
+ },
460
+ {
461
+ "epoch": 0.70234050226837,
462
+ "grad_norm": 0.10302391648292542,
463
+ "learning_rate": 0.00023441880547331984,
464
+ "loss": 1.209429443359375,
465
+ "step": 18500
466
+ },
467
+ {
468
+ "epoch": 0.721322678005353,
469
+ "grad_norm": 0.16614890098571777,
470
+ "learning_rate": 0.00023044406843431085,
471
+ "loss": 1.210209228515625,
472
+ "step": 19000
473
+ },
474
+ {
475
+ "epoch": 0.7403048537423359,
476
+ "grad_norm": 0.12630845606327057,
477
+ "learning_rate": 0.00022638839971544675,
478
+ "loss": 1.2063096923828125,
479
+ "step": 19500
480
+ },
481
+ {
482
+ "epoch": 0.7592870294793189,
483
+ "grad_norm": 0.16276787221431732,
484
+ "learning_rate": 0.00022225587956886718,
485
+ "loss": 1.20748486328125,
486
+ "step": 20000
487
+ },
488
+ {
489
+ "epoch": 0.7592870294793189,
490
+ "eval_correction_accuracy": 0.974335110021822,
491
+ "eval_detection_f1": 0.6143125770798066,
492
+ "eval_loss": 1.234763264656067,
493
+ "eval_runtime": 60.5996,
494
+ "eval_samples_per_second": 370.332,
495
+ "eval_steps_per_second": 2.904,
496
+ "step": 20000
497
+ },
498
+ {
499
+ "epoch": 0.7592870294793189,
500
+ "eval_edit_f0_5": 0.9024403864410565,
501
+ "eval_edit_f0_5_m0": 0.8842469830618539,
502
+ "eval_edit_f0_5_m0.1": 0.8886223627541842,
503
+ "eval_edit_f0_5_m0.2": 0.8945219837810999,
504
+ "eval_edit_f0_5_m0.3": 0.9024403864410565,
505
+ "step": 20000
506
+ },
507
+ {
508
+ "epoch": 0.7592870294793189,
509
+ "eval_real_f0_5": 0.4630195599022005,
510
+ "eval_real_f0_5_m0": 0.44520547945205474,
511
+ "eval_real_f0_5_m0.1": 0.4467213114754099,
512
+ "eval_real_f0_5_m0.2": 0.4604503464203233,
513
+ "eval_real_f0_5_m0.3": 0.4630195599022005,
514
+ "step": 20000
515
+ },
516
+ {
517
+ "epoch": 0.7782692052163019,
518
+ "grad_norm": 0.16368603706359863,
519
+ "learning_rate": 0.0002180506655639742,
520
+ "loss": 1.20710107421875,
521
+ "step": 20500
522
+ },
523
+ {
524
+ "epoch": 0.7972513809532848,
525
+ "grad_norm": 0.1419440507888794,
526
+ "learning_rate": 0.00021377698840466171,
527
+ "loss": 1.207209716796875,
528
+ "step": 21000
529
+ },
530
+ {
531
+ "epoch": 0.8162335566902679,
532
+ "grad_norm": 0.2440534383058548,
533
+ "learning_rate": 0.00020943914767296716,
534
+ "loss": 1.203267333984375,
535
+ "step": 21500
536
+ },
537
+ {
538
+ "epoch": 0.8352157324272508,
539
+ "grad_norm": 0.11360731720924377,
540
+ "learning_rate": 0.00020504150750342767,
541
+ "loss": 1.2064322509765626,
542
+ "step": 22000
543
+ },
544
+ {
545
+ "epoch": 0.8541979081642338,
546
+ "grad_norm": 0.19054335355758667,
547
+ "learning_rate": 0.00020058849219249225,
548
+ "loss": 1.2044508056640626,
549
+ "step": 22500
550
+ },
551
+ {
552
+ "epoch": 0.8541979081642338,
553
+ "eval_correction_accuracy": 0.9746169932390887,
554
+ "eval_detection_f1": 0.6164803461954562,
555
+ "eval_loss": 1.2325071096420288,
556
+ "eval_runtime": 60.5986,
557
+ "eval_samples_per_second": 370.339,
558
+ "eval_steps_per_second": 2.904,
559
+ "step": 22500
560
+ },
561
+ {
562
+ "epoch": 0.8541979081642338,
563
+ "eval_edit_f0_5": 0.9041808921475346,
564
+ "eval_edit_f0_5_m0": 0.8936088786835056,
565
+ "eval_edit_f0_5_m0.1": 0.8979367262723521,
566
+ "eval_edit_f0_5_m0.2": 0.9006915629322267,
567
+ "eval_edit_f0_5_m0.3": 0.9041808921475346,
568
+ "step": 22500
569
+ },
570
+ {
571
+ "epoch": 0.8541979081642338,
572
+ "eval_real_f0_5": 0.43545659526493796,
573
+ "eval_real_f0_5_m0": 0.4005456349206349,
574
+ "eval_real_f0_5_m0.1": 0.4166666666666667,
575
+ "eval_real_f0_5_m0.2": 0.4315960912052117,
576
+ "eval_real_f0_5_m0.3": 0.43545659526493796,
577
+ "step": 22500
578
+ },
579
+ {
580
+ "epoch": 0.8731800839012167,
581
+ "grad_norm": 0.15163150429725647,
582
+ "learning_rate": 0.00019608458174740764,
583
+ "loss": 1.205394287109375,
584
+ "step": 23000
585
+ },
586
+ {
587
+ "epoch": 0.8921622596381997,
588
+ "grad_norm": 0.1204524040222168,
589
+ "learning_rate": 0.00019153430737905516,
590
+ "loss": 1.20386767578125,
591
+ "step": 23500
592
+ },
593
+ {
594
+ "epoch": 0.9111444353751827,
595
+ "grad_norm": 0.14975769817829132,
596
+ "learning_rate": 0.00018694224694327434,
597
+ "loss": 1.2036343994140626,
598
+ "step": 24000
599
+ },
600
+ {
601
+ "epoch": 0.9301266111121657,
602
+ "grad_norm": 0.09973002225160599,
603
+ "learning_rate": 0.0001823130203352581,
604
+ "loss": 1.202430419921875,
605
+ "step": 24500
606
+ },
607
+ {
608
+ "epoch": 0.9491087868491487,
609
+ "grad_norm": 0.12494324147701263,
610
+ "learning_rate": 0.00017765128484165446,
611
+ "loss": 1.203585205078125,
612
+ "step": 25000
613
+ },
614
+ {
615
+ "epoch": 0.9491087868491487,
616
+ "eval_correction_accuracy": 0.9748879967532328,
617
+ "eval_detection_f1": 0.6170252572497661,
618
+ "eval_loss": 1.2309163808822632,
619
+ "eval_runtime": 59.979,
620
+ "eval_samples_per_second": 374.164,
621
+ "eval_steps_per_second": 2.934,
622
+ "step": 25000
623
+ },
624
+ {
625
+ "epoch": 0.9491087868491487,
626
+ "eval_edit_f0_5": 0.9054563883101525,
627
+ "eval_edit_f0_5_m0": 0.8931397696883698,
628
+ "eval_edit_f0_5_m0.1": 0.8968065743753791,
629
+ "eval_edit_f0_5_m0.2": 0.9016347709646821,
630
+ "eval_edit_f0_5_m0.3": 0.9054563883101525,
631
+ "step": 25000
632
+ },
633
+ {
634
+ "epoch": 0.9491087868491487,
635
+ "eval_real_f0_5": 0.46703296703296693,
636
+ "eval_real_f0_5_m0": 0.43244626407369496,
637
+ "eval_real_f0_5_m0.1": 0.44786096256684493,
638
+ "eval_real_f0_5_m0.2": 0.45962199312714774,
639
+ "eval_real_f0_5_m0.3": 0.46703296703296693,
640
+ "step": 25000
641
+ },
642
+ {
643
+ "epoch": 0.9680909625861316,
644
+ "grad_norm": 0.14670301973819733,
645
+ "learning_rate": 0.00017296173045504914,
646
+ "loss": 1.2033277587890625,
647
+ "step": 25500
648
+ },
649
+ {
650
+ "epoch": 0.9870731383231146,
651
+ "grad_norm": 0.10602114349603653,
652
+ "learning_rate": 0.00016824907515554526,
653
+ "loss": 1.201916748046875,
654
+ "step": 26000
655
+ },
656
+ {
657
+ "epoch": 1.0060363318843606,
658
+ "grad_norm": 0.06282790750265121,
659
+ "learning_rate": 0.00016351806016418486,
660
+ "loss": 1.199654541015625,
661
+ "step": 26500
662
+ },
663
+ {
664
+ "epoch": 1.0250185076213436,
665
+ "grad_norm": 0.06489919126033783,
666
+ "learning_rate": 0.00015877344517298893,
667
+ "loss": 1.19592431640625,
668
+ "step": 27000
669
+ },
670
+ {
671
+ "epoch": 1.0440006833583266,
672
+ "grad_norm": 0.06663327664136887,
673
+ "learning_rate": 0.0001540200035564149,
674
+ "loss": 1.19781689453125,
675
+ "step": 27500
676
+ },
677
+ {
678
+ "epoch": 1.0440006833583266,
679
+ "eval_correction_accuracy": 0.9751896612489042,
680
+ "eval_detection_f1": 0.6337458056595326,
681
+ "eval_loss": 1.2278345823287964,
682
+ "eval_runtime": 60.4482,
683
+ "eval_samples_per_second": 371.26,
684
+ "eval_steps_per_second": 2.912,
685
+ "step": 27500
686
+ },
687
+ {
688
+ "epoch": 1.0440006833583266,
689
+ "eval_edit_f0_5": 0.9052714133661981,
690
+ "eval_edit_f0_5_m0": 0.8921670023406456,
691
+ "eval_edit_f0_5_m0.1": 0.896949711459192,
692
+ "eval_edit_f0_5_m0.2": 0.9007485444968119,
693
+ "eval_edit_f0_5_m0.3": 0.9052714133661981,
694
+ "step": 27500
695
+ },
696
+ {
697
+ "epoch": 1.0440006833583266,
698
+ "eval_real_f0_5": 0.47462941847206386,
699
+ "eval_real_f0_5_m0": 0.4401750972762646,
700
+ "eval_real_f0_5_m0.1": 0.46072895277207393,
701
+ "eval_real_f0_5_m0.2": 0.4667207792207792,
702
+ "eval_real_f0_5_m0.3": 0.47462941847206386,
703
+ "step": 27500
704
+ },
705
+ {
706
+ "epoch": 1.0629828590953094,
707
+ "grad_norm": 0.06707123667001724,
708
+ "learning_rate": 0.0001492625175690478,
709
+ "loss": 1.196161865234375,
710
+ "step": 28000
711
+ },
712
+ {
713
+ "epoch": 1.0819650348322924,
714
+ "grad_norm": 0.04345293343067169,
715
+ "learning_rate": 0.00014450577353435873,
716
+ "loss": 1.1952852783203125,
717
+ "step": 28500
718
+ },
719
+ {
720
+ "epoch": 1.1009472105692755,
721
+ "grad_norm": 0.03876705467700958,
722
+ "learning_rate": 0.00013975455702936845,
723
+ "loss": 1.19596826171875,
724
+ "step": 29000
725
+ },
726
+ {
727
+ "epoch": 1.1199293863062585,
728
+ "grad_norm": 0.054508332163095474,
729
+ "learning_rate": 0.00013501364807006326,
730
+ "loss": 1.1936068115234375,
731
+ "step": 29500
732
+ },
733
+ {
734
+ "epoch": 1.1389115620432415,
735
+ "grad_norm": 0.05647437274456024,
736
+ "learning_rate": 0.0001302878163024043,
737
+ "loss": 1.19433984375,
738
+ "step": 30000
739
+ },
740
+ {
741
+ "epoch": 1.1389115620432415,
742
+ "eval_correction_accuracy": 0.9752921942086351,
743
+ "eval_detection_f1": 0.6477482980155775,
744
+ "eval_loss": 1.2278577089309692,
745
+ "eval_runtime": 59.9225,
746
+ "eval_samples_per_second": 374.517,
747
+ "eval_steps_per_second": 2.937,
748
+ "step": 30000
749
+ },
750
+ {
751
+ "epoch": 1.1389115620432415,
752
+ "eval_edit_f0_5": 0.9029944709038156,
753
+ "eval_edit_f0_5_m0": 0.8870581982460802,
754
+ "eval_edit_f0_5_m0.1": 0.8917932478127851,
755
+ "eval_edit_f0_5_m0.2": 0.8975887293416418,
756
+ "eval_edit_f0_5_m0.3": 0.9029944709038156,
757
+ "step": 30000
758
+ },
759
+ {
760
+ "epoch": 1.1389115620432415,
761
+ "eval_real_f0_5": 0.46045785639958375,
762
+ "eval_real_f0_5_m0": 0.411169744942832,
763
+ "eval_real_f0_5_m0.1": 0.4227605118829982,
764
+ "eval_real_f0_5_m0.2": 0.4366602687140115,
765
+ "eval_real_f0_5_m0.3": 0.46045785639958375,
766
+ "step": 30000
767
+ },
768
+ {
769
+ "epoch": 1.1578937377802243,
770
+ "grad_norm": 0.08290459215641022,
771
+ "learning_rate": 0.00012558181620377085,
772
+ "loss": 1.196980224609375,
773
+ "step": 30500
774
+ },
775
+ {
776
+ "epoch": 1.1768759135172073,
777
+ "grad_norm": 0.03389839455485344,
778
+ "learning_rate": 0.00012090038229966375,
779
+ "loss": 1.19385400390625,
780
+ "step": 31000
781
+ },
782
+ {
783
+ "epoch": 1.1958580892541903,
784
+ "grad_norm": 0.03219689056277275,
785
+ "learning_rate": 0.0001162482244004811,
786
+ "loss": 1.1932216796875,
787
+ "step": 31500
788
+ },
789
+ {
790
+ "epoch": 1.2148402649911734,
791
+ "grad_norm": 0.03629528358578682,
792
+ "learning_rate": 0.00011163002286315993,
793
+ "loss": 1.1932420654296876,
794
+ "step": 32000
795
+ },
796
+ {
797
+ "epoch": 1.2338224407281562,
798
+ "grad_norm": 0.05208583176136017,
799
+ "learning_rate": 0.00010705042388244893,
800
+ "loss": 1.1948599853515625,
801
+ "step": 32500
802
+ },
803
+ {
804
+ "epoch": 1.2338224407281562,
805
+ "eval_correction_accuracy": 0.9756340806370627,
806
+ "eval_detection_f1": 0.6500135636133466,
807
+ "eval_loss": 1.2252331972122192,
808
+ "eval_runtime": 60.3519,
809
+ "eval_samples_per_second": 371.852,
810
+ "eval_steps_per_second": 2.916,
811
+ "step": 32500
812
+ },
813
+ {
814
+ "epoch": 1.2338224407281562,
815
+ "eval_edit_f0_5": 0.9102684230832827,
816
+ "eval_edit_f0_5_m0": 0.8981327019318732,
817
+ "eval_edit_f0_5_m0.1": 0.9021497806898794,
818
+ "eval_edit_f0_5_m0.2": 0.9072063717200373,
819
+ "eval_edit_f0_5_m0.3": 0.9102684230832827,
820
+ "step": 32500
821
+ },
822
+ {
823
+ "epoch": 1.2338224407281562,
824
+ "eval_real_f0_5": 0.4674586776859504,
825
+ "eval_real_f0_5_m0": 0.4159212880143113,
826
+ "eval_real_f0_5_m0.1": 0.4346804511278195,
827
+ "eval_real_f0_5_m0.2": 0.4519704433497537,
828
+ "eval_real_f0_5_m0.3": 0.4674586776859504,
829
+ "step": 32500
830
+ },
831
+ {
832
+ "epoch": 1.2528046164651392,
833
+ "grad_norm": 0.06928142160177231,
834
+ "learning_rate": 0.00010251403481655096,
835
+ "loss": 1.1938966064453125,
836
+ "step": 33000
837
+ },
838
+ {
839
+ "epoch": 1.2717867922021222,
840
+ "grad_norm": 0.05749162659049034,
841
+ "learning_rate": 9.80254195518373e-05,
842
+ "loss": 1.1921290283203125,
843
+ "step": 33500
844
+ },
845
+ {
846
+ "epoch": 1.2907689679391052,
847
+ "grad_norm": 0.04986289143562317,
848
+ "learning_rate": 9.358909391129737e-05,
849
+ "loss": 1.19255517578125,
850
+ "step": 34000
851
+ },
852
+ {
853
+ "epoch": 1.3097511436760882,
854
+ "grad_norm": 0.1535189300775528,
855
+ "learning_rate": 8.920952111134303e-05,
856
+ "loss": 1.19104052734375,
857
+ "step": 34500
858
+ },
859
+ {
860
+ "epoch": 1.328733319413071,
861
+ "grad_norm": 0.042191170156002045,
862
+ "learning_rate": 8.489110727153841e-05,
863
+ "loss": 1.191498779296875,
864
+ "step": 35000
865
+ },
866
+ {
867
+ "epoch": 1.328733319413071,
868
+ "eval_correction_accuracy": 0.9758862919367224,
869
+ "eval_detection_f1": 0.6438216536684808,
870
+ "eval_loss": 1.2241498231887817,
871
+ "eval_runtime": 59.6334,
872
+ "eval_samples_per_second": 376.333,
873
+ "eval_steps_per_second": 2.951,
874
+ "step": 35000
875
+ },
876
+ {
877
+ "epoch": 1.328733319413071,
878
+ "eval_edit_f0_5": 0.9100482181455414,
879
+ "eval_edit_f0_5_m0": 0.9021638917511796,
880
+ "eval_edit_f0_5_m0.1": 0.9037821317469846,
881
+ "eval_edit_f0_5_m0.2": 0.9073168050134681,
882
+ "eval_edit_f0_5_m0.3": 0.9100482181455414,
883
+ "step": 35000
884
+ },
885
+ {
886
+ "epoch": 1.328733319413071,
887
+ "eval_real_f0_5": 0.4544167610419026,
888
+ "eval_real_f0_5_m0": 0.4211689587426326,
889
+ "eval_real_f0_5_m0.1": 0.42387983706720966,
890
+ "eval_real_f0_5_m0.2": 0.4281914893617021,
891
+ "eval_real_f0_5_m0.3": 0.4544167610419026,
892
+ "step": 35000
893
+ },
894
+ {
895
+ "epoch": 1.347715495150054,
896
+ "grad_norm": 0.029605668038129807,
897
+ "learning_rate": 8.063819698177278e-05,
898
+ "loss": 1.192787841796875,
899
+ "step": 35500
900
+ },
901
+ {
902
+ "epoch": 1.366697670887037,
903
+ "grad_norm": 0.048531949520111084,
904
+ "learning_rate": 7.645506893133574e-05,
905
+ "loss": 1.190103759765625,
906
+ "step": 36000
907
+ },
908
+ {
909
+ "epoch": 1.38567984662402,
910
+ "grad_norm": 0.03906732425093651,
911
+ "learning_rate": 7.234593160429277e-05,
912
+ "loss": 1.18997265625,
913
+ "step": 36500
914
+ },
915
+ {
916
+ "epoch": 1.404662022361003,
917
+ "grad_norm": 0.03661827743053436,
918
+ "learning_rate": 6.831491904549145e-05,
919
+ "loss": 1.191602294921875,
920
+ "step": 37000
921
+ },
922
+ {
923
+ "epoch": 1.423644198097986,
924
+ "grad_norm": 0.043553389608860016,
925
+ "learning_rate": 6.436608670145828e-05,
926
+ "loss": 1.1902423095703125,
927
+ "step": 37500
928
+ },
929
+ {
930
+ "epoch": 1.423644198097986,
931
+ "eval_correction_accuracy": 0.9760330030848904,
932
+ "eval_detection_f1": 0.6487232574189096,
933
+ "eval_loss": 1.222518801689148,
934
+ "eval_runtime": 60.4683,
935
+ "eval_samples_per_second": 371.137,
936
+ "eval_steps_per_second": 2.911,
937
+ "step": 37500
938
+ },
939
+ {
940
+ "epoch": 1.423644198097986,
941
+ "eval_edit_f0_5": 0.9084162497938542,
942
+ "eval_edit_f0_5_m0": 0.8946242310778283,
943
+ "eval_edit_f0_5_m0.1": 0.9003717872730211,
944
+ "eval_edit_f0_5_m0.2": 0.9063674492074731,
945
+ "eval_edit_f0_5_m0.3": 0.9084162497938542,
946
+ "step": 37500
947
+ },
948
+ {
949
+ "epoch": 1.423644198097986,
950
+ "eval_real_f0_5": 0.44614147909967844,
951
+ "eval_real_f0_5_m0": 0.4033728350045579,
952
+ "eval_real_f0_5_m0.1": 0.4206963249516441,
953
+ "eval_real_f0_5_m0.2": 0.4333333333333333,
954
+ "eval_real_f0_5_m0.3": 0.44614147909967844,
955
+ "step": 37500
956
+ },
957
+ {
958
+ "epoch": 1.442626373834969,
959
+ "grad_norm": 0.045576661825180054,
960
+ "learning_rate": 6.050340734036973e-05,
961
+ "loss": 1.1904107666015624,
962
+ "step": 38000
963
+ },
964
+ {
965
+ "epoch": 1.461608549571952,
966
+ "grad_norm": 0.04765897989273071,
967
+ "learning_rate": 5.673076705520346e-05,
968
+ "loss": 1.1911429443359376,
969
+ "step": 38500
970
+ },
971
+ {
972
+ "epoch": 1.480590725308935,
973
+ "grad_norm": 0.10767372697591782,
974
+ "learning_rate": 5.3051961354089736e-05,
975
+ "loss": 1.1882059326171874,
976
+ "step": 39000
977
+ },
978
+ {
979
+ "epoch": 1.499572901045918,
980
+ "grad_norm": 0.036085955798625946,
981
+ "learning_rate": 4.94706913417966e-05,
982
+ "loss": 1.190911376953125,
983
+ "step": 39500
984
+ },
985
+ {
986
+ "epoch": 1.5185550767829008,
987
+ "grad_norm": 0.08071824908256531,
988
+ "learning_rate": 4.599055999619126e-05,
989
+ "loss": 1.1896280517578126,
990
+ "step": 40000
991
+ },
992
+ {
993
+ "epoch": 1.5185550767829008,
994
+ "eval_correction_accuracy": 0.976151361067345,
995
+ "eval_detection_f1": 0.6554347875581136,
996
+ "eval_loss": 1.2218546867370605,
997
+ "eval_runtime": 60.355,
998
+ "eval_samples_per_second": 371.833,
999
+ "eval_steps_per_second": 2.916,
1000
+ "step": 40000
1001
+ },
1002
+ {
1003
+ "epoch": 1.5185550767829008,
1004
+ "eval_edit_f0_5": 0.910992458853966,
1005
+ "eval_edit_f0_5_m0": 0.9010734128054373,
1006
+ "eval_edit_f0_5_m0.1": 0.9040790831568084,
1007
+ "eval_edit_f0_5_m0.2": 0.9071533963604569,
1008
+ "eval_edit_f0_5_m0.3": 0.910992458853966,
1009
+ "step": 40000
1010
+ },
1011
+ {
1012
+ "epoch": 1.5185550767829008,
1013
+ "eval_real_f0_5": 0.44711014176663033,
1014
+ "eval_real_f0_5_m0": 0.4186210131332083,
1015
+ "eval_real_f0_5_m0.1": 0.4342492639842983,
1016
+ "eval_real_f0_5_m0.2": 0.4353493222106361,
1017
+ "eval_real_f0_5_m0.3": 0.44711014176663033,
1018
+ "step": 40000
1019
+ },
1020
+ {
1021
+ "epoch": 1.5375372525198838,
1022
+ "grad_norm": 0.030372846871614456,
1023
+ "learning_rate": 4.2615068543422434e-05,
1024
+ "loss": 1.19057763671875,
1025
+ "step": 40500
1026
+ },
1027
+ {
1028
+ "epoch": 1.5565194282568668,
1029
+ "grad_norm": 0.03362102061510086,
1030
+ "learning_rate": 3.934761293547178e-05,
1031
+ "loss": 1.1867962646484376,
1032
+ "step": 41000
1033
+ },
1034
+ {
1035
+ "epoch": 1.5755016039938496,
1036
+ "grad_norm": 0.057195816189050674,
1037
+ "learning_rate": 3.619148043361715e-05,
1038
+ "loss": 1.1873057861328125,
1039
+ "step": 41500
1040
+ },
1041
+ {
1042
+ "epoch": 1.5944837797308327,
1043
+ "grad_norm": 0.03772829845547676,
1044
+ "learning_rate": 3.314984630124587e-05,
1045
+ "loss": 1.1879420166015624,
1046
+ "step": 42000
1047
+ },
1048
+ {
1049
+ "epoch": 1.6134659554678157,
1050
+ "grad_norm": 0.1718292534351349,
1051
+ "learning_rate": 3.022577060934427e-05,
1052
+ "loss": 1.1901536865234374,
1053
+ "step": 42500
1054
+ },
1055
+ {
1056
+ "epoch": 1.6134659554678157,
1057
+ "eval_correction_accuracy": 0.9762463112036875,
1058
+ "eval_detection_f1": 0.6603370175756478,
1059
+ "eval_loss": 1.2204469442367554,
1060
+ "eval_runtime": 59.5936,
1061
+ "eval_samples_per_second": 376.584,
1062
+ "eval_steps_per_second": 2.953,
1063
+ "step": 42500
1064
+ },
1065
+ {
1066
+ "epoch": 1.6134659554678157,
1067
+ "eval_edit_f0_5": 0.9098653476229733,
1068
+ "eval_edit_f0_5_m0": 0.8979657559980677,
1069
+ "eval_edit_f0_5_m0.1": 0.9045865598093898,
1070
+ "eval_edit_f0_5_m0.2": 0.9075277034772642,
1071
+ "eval_edit_f0_5_m0.3": 0.9098653476229733,
1072
+ "step": 42500
1073
+ },
1074
+ {
1075
+ "epoch": 1.6134659554678157,
1076
+ "eval_real_f0_5": 0.44262720664589816,
1077
+ "eval_real_f0_5_m0": 0.42281105990783413,
1078
+ "eval_real_f0_5_m0.1": 0.4292809839167455,
1079
+ "eval_real_f0_5_m0.2": 0.43297455968688847,
1080
+ "eval_real_f0_5_m0.3": 0.44262720664589816,
1081
+ "step": 42500
1082
+ },
1083
+ {
1084
+ "epoch": 1.6324481312047987,
1085
+ "grad_norm": 0.027738308534026146,
1086
+ "learning_rate": 2.742219515787808e-05,
1087
+ "loss": 1.186643798828125,
1088
+ "step": 43000
1089
+ },
1090
+ {
1091
+ "epoch": 1.6514303069417817,
1092
+ "grad_norm": 0.03820585086941719,
1093
+ "learning_rate": 2.4741940516161014e-05,
1094
+ "loss": 1.1873551025390625,
1095
+ "step": 43500
1096
+ },
1097
+ {
1098
+ "epoch": 1.6704124826787647,
1099
+ "grad_norm": 0.05109170451760292,
1100
+ "learning_rate": 2.2187703185188212e-05,
1101
+ "loss": 1.1882010498046875,
1102
+ "step": 44000
1103
+ },
1104
+ {
1105
+ "epoch": 1.6893946584157478,
1106
+ "grad_norm": 0.04304545372724533,
1107
+ "learning_rate": 1.9762052884790763e-05,
1108
+ "loss": 1.18957080078125,
1109
+ "step": 44500
1110
+ },
1111
+ {
1112
+ "epoch": 1.7083768341527306,
1113
+ "grad_norm": 0.05135943740606308,
1114
+ "learning_rate": 1.7467429968339254e-05,
1115
+ "loss": 1.1895230712890625,
1116
+ "step": 45000
1117
+ },
1118
+ {
1119
+ "epoch": 1.7083768341527306,
1120
+ "eval_correction_accuracy": 0.9763739004493976,
1121
+ "eval_detection_f1": 0.6614078220399721,
1122
+ "eval_loss": 1.220071792602539,
1123
+ "eval_runtime": 60.3321,
1124
+ "eval_samples_per_second": 371.975,
1125
+ "eval_steps_per_second": 2.917,
1126
+ "step": 45000
1127
+ },
1128
+ {
1129
+ "epoch": 1.7083768341527306,
1130
+ "eval_edit_f0_5": 0.912360661138872,
1131
+ "eval_edit_f0_5_m0": 0.9017536332922186,
1132
+ "eval_edit_f0_5_m0.1": 0.9063933099541408,
1133
+ "eval_edit_f0_5_m0.2": 0.9094366273700224,
1134
+ "eval_edit_f0_5_m0.3": 0.912360661138872,
1135
+ "step": 45000
1136
+ },
1137
+ {
1138
+ "epoch": 1.7083768341527306,
1139
+ "eval_real_f0_5": 0.44325313807531386,
1140
+ "eval_real_f0_5_m0": 0.4201485608170845,
1141
+ "eval_real_f0_5_m0.1": 0.42870905587668595,
1142
+ "eval_real_f0_5_m0.2": 0.4367469879518072,
1143
+ "eval_real_f0_5_m0.3": 0.44325313807531386,
1144
+ "step": 45000
1145
+ },
1146
+ {
1147
+ "epoch": 1.7273590098897136,
1148
+ "grad_norm": 0.03698679432272911,
1149
+ "learning_rate": 1.5306142967598462e-05,
1150
+ "loss": 1.1872025146484375,
1151
+ "step": 45500
1152
+ },
1153
+ {
1154
+ "epoch": 1.7463411856266964,
1155
+ "grad_norm": 0.043697986751794815,
1156
+ "learning_rate": 1.3280366270202187e-05,
1157
+ "loss": 1.1902381591796876,
1158
+ "step": 46000
1159
+ },
1160
+ {
1161
+ "epoch": 1.7653233613636794,
1162
+ "grad_norm": 0.0478336326777935,
1163
+ "learning_rate": 1.1392137932085692e-05,
1164
+ "loss": 1.188957763671875,
1165
+ "step": 46500
1166
+ },
1167
+ {
1168
+ "epoch": 1.7843055371006624,
1169
+ "grad_norm": 0.04866790026426315,
1170
+ "learning_rate": 9.643357627076176e-06,
1171
+ "loss": 1.188841064453125,
1172
+ "step": 47000
1173
+ },
1174
+ {
1175
+ "epoch": 1.8032877128376454,
1176
+ "grad_norm": 0.08654382824897766,
1177
+ "learning_rate": 8.035784735703915e-06,
1178
+ "loss": 1.1869093017578125,
1179
+ "step": 47500
1180
+ },
1181
+ {
1182
+ "epoch": 1.8032877128376454,
1183
+ "eval_correction_accuracy": 0.9764408271080002,
1184
+ "eval_detection_f1": 0.6623505252062661,
1185
+ "eval_loss": 1.2196118831634521,
1186
+ "eval_runtime": 60.2937,
1187
+ "eval_samples_per_second": 372.211,
1188
+ "eval_steps_per_second": 2.919,
1189
+ "step": 47500
1190
+ },
1191
+ {
1192
+ "epoch": 1.8032877128376454,
1193
+ "eval_edit_f0_5": 0.9131843391356872,
1194
+ "eval_edit_f0_5_m0": 0.901242536711312,
1195
+ "eval_edit_f0_5_m0.1": 0.9049602423324499,
1196
+ "eval_edit_f0_5_m0.2": 0.9078295881168726,
1197
+ "eval_edit_f0_5_m0.3": 0.9131843391356872,
1198
+ "step": 47500
1199
+ },
1200
+ {
1201
+ "epoch": 1.8032877128376454,
1202
+ "eval_real_f0_5": 0.4527455121436114,
1203
+ "eval_real_f0_5_m0": 0.4291393826005612,
1204
+ "eval_real_f0_5_m0.1": 0.4339440694310511,
1205
+ "eval_real_f0_5_m0.2": 0.4473161033797216,
1206
+ "eval_real_f0_5_m0.3": 0.4527455121436114,
1207
+ "step": 47500
1208
+ },
1209
+ {
1210
+ "epoch": 1.8222698885746285,
1211
+ "grad_norm": 0.07231193780899048,
1212
+ "learning_rate": 6.571036575157384e-06,
1213
+ "loss": 1.187695556640625,
1214
+ "step": 48000
1215
+ },
1216
+ {
1217
+ "epoch": 1.8412520643116115,
1218
+ "grad_norm": 0.04208879545331001,
1219
+ "learning_rate": 5.250586772162613e-06,
1220
+ "loss": 1.1868779296875,
1221
+ "step": 48500
1222
+ },
1223
+ {
1224
+ "epoch": 1.8602342400485945,
1225
+ "grad_norm": 0.08146563172340393,
1226
+ "learning_rate": 4.07576378042419e-06,
1227
+ "loss": 1.1862469482421876,
1228
+ "step": 49000
1229
+ },
1230
+ {
1231
+ "epoch": 1.8792164157855773,
1232
+ "grad_norm": 0.035731397569179535,
1233
+ "learning_rate": 3.047749544119016e-06,
1234
+ "loss": 1.1868427734375,
1235
+ "step": 49500
1236
+ },
1237
+ {
1238
+ "epoch": 1.8981985915225603,
1239
+ "grad_norm": 0.049494218081235886,
1240
+ "learning_rate": 2.167578308787865e-06,
1241
+ "loss": 1.18650390625,
1242
+ "step": 50000
1243
+ },
1244
+ {
1245
+ "epoch": 1.8981985915225603,
1246
+ "eval_correction_accuracy": 0.9764698396496604,
1247
+ "eval_detection_f1": 0.662213053912394,
1248
+ "eval_loss": 1.219179391860962,
1249
+ "eval_runtime": 60.3298,
1250
+ "eval_samples_per_second": 371.989,
1251
+ "eval_steps_per_second": 2.917,
1252
+ "step": 50000
1253
+ },
1254
+ {
1255
+ "epoch": 1.8981985915225603,
1256
+ "eval_edit_f0_5": 0.9142088207832153,
1257
+ "eval_edit_f0_5_m0": 0.904126540054877,
1258
+ "eval_edit_f0_5_m0.1": 0.9078098748035337,
1259
+ "eval_edit_f0_5_m0.2": 0.9097596861206475,
1260
+ "eval_edit_f0_5_m0.3": 0.9142088207832153,
1261
+ "step": 50000
1262
+ },
1263
+ {
1264
+ "epoch": 1.8981985915225603,
1265
+ "eval_real_f0_5": 0.44823906083244397,
1266
+ "eval_real_f0_5_m0": 0.4280018761726078,
1267
+ "eval_real_f0_5_m0.1": 0.43232177263969174,
1268
+ "eval_real_f0_5_m0.2": 0.4433867735470942,
1269
+ "eval_real_f0_5_m0.3": 0.44823906083244397,
1270
+ "step": 50000
1271
+ }
1272
+ ],
1273
+ "logging_steps": 500,
1274
+ "max_steps": 52682,
1275
+ "num_input_tokens_seen": 0,
1276
+ "num_train_epochs": 2,
1277
+ "save_steps": 2500,
1278
+ "stateful_callbacks": {
1279
+ "TrainerControl": {
1280
+ "args": {
1281
+ "should_epoch_stop": false,
1282
+ "should_evaluate": false,
1283
+ "should_log": false,
1284
+ "should_save": true,
1285
+ "should_training_stop": false
1286
+ },
1287
+ "attributes": {}
1288
+ }
1289
+ },
1290
+ "total_flos": 1.1632343079571684e+18,
1291
+ "train_batch_size": 32,
1292
+ "trial_name": null,
1293
+ "trial_params": null
1294
+ }
checkpoint-50000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae772d11cb059004c86012d10f5afc5062b15c0c8ba45184d07ee30411bd6a0a
3
+ size 5368
checkpoint-50000/vocab.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tokens": ["<pad>", "<bos>", "<eos>", "<unk>", "\u0003", "\t", "\n", "\f", "\r", "\u0012", "\u0015", " ", "!", "\"", "#", "$", "%", "&", "'", "(", ")", "*", "+", ",", "-", ".", "/", "0", "1", "2", "3", "4", "5", "6", "7", "8", "9", ":", ";", "<", "=", ">", "?", "@", "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z", "[", "\\", "]", "^", "_", "`", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "{", "|", "}", "~", "€", "ƒ", "", "‘", "’", "“", "”", "•", "–", "—", "˜", "™", "œ", "", " ", "¡", "¢", "£", "¤", "¥", "¦", "§", "¨", "©", "«", "¬", "­", "®", "¯", "°", "±", "²", "³", "´", "µ", "¶", "·", "¹", "º", "»", "¼", "½", "¾", "¿", "À", "Á", "Â", "Ã", "Ä", "Å", "Æ", "Ç", "È", "É", "Ê", "Í", "Î", "Ð", "Ó", "Õ", "Ö", "×", "Ø", "Ù", "Ú", "Û", "Ü", "Ý", "Þ", "ß", "à", "á", "â", "ã", "ä", "å", "æ", "ç", "è", "é", "ê", "ë", "ì", "í", "î", "ï", "ð", "ñ", "ò", "ó", "ô", "õ", "ö", "ø", "ù", "ú", "û", "ü", "ý", "þ", "ÿ", "Ā", "ā", "ă", "ą", "Ć", "ć", "ċ", "Č", "č", "ď", "Đ", "đ", "ē", "ę", "ě", "Ğ", "ğ", "Ģ", "ģ", "ĩ", "ī", "ĭ", "İ", "ı", "Ķ", "ķ", "ļ", "Ľ", "Ł", "ł", "ń", "ņ", "ň", "Ŋ", "ŋ", "Ō", "ō", "Ŏ", "ŏ", "ő", "Œ", "œ", "Ř", "ř", "Ś", "ś", "Ş", "ş", "Š", "š", "ţ", "Ũ", "Ū", "ū", "ŭ", "ů", "ű", "ż", "Ž", "ž", "ƒ", "Ɲ", "ư", "ƴ", "ǎ", "ǒ", "ș", "ȝ", "ɑ", "ɓ", "ɔ", "ɕ", "ɖ", "ɘ", "ə", "ɛ", "ɡ", "ɣ", "ɧ", "ɪ", "ɬ", "ɯ", "ɲ", "ʀ", "ʁ", "ʃ", "ʉ", "ʊ", "ʋ", "ʌ", "ʏ", "ʔ", "ʕ", "ʖ", "ʰ", "ʻ", "ʿ", "ˁ", "ˆ", "ˈ", "ˌ", "ˎ", "ː", "˚", "˜", "˟", "̀", "́", "̂", "̃", "̅", "̈", "̌", "̥", "̯", "͜", "͡", "ͼ", "Ό", "Α", "Γ", "Δ", "Ε", "Λ", "Ν", "Ξ", "Π", "Τ", "Φ", "Ψ", "Ω", "ά", "έ", "ή", "ί", "α", "β", "γ", "δ", "ε", "ζ", "η", "θ", "ι", "κ", "λ", "μ", "ν", "ο", "π", "ρ", "ς", "σ", "τ", "υ", "φ", "χ", "ψ", "ω", "ό", "ύ", "ώ", "ϒ", "Ђ", "І", "А", "Б", "В", "Г", "Д", "И", "К", "М", "Н", "П", "Р", "С", "Т", "Ф", "Ц", "Ч", "Ш", "Ю", "Я", "а", "б", "в", "г", "д", "е", "ж", "з", "и", "й", "к", "л", "м", "н", "о", "п", "р", "с", "т", "у", "ф", "х", "ц", "ч", "ш", "щ", "ы", "ь", "ю", "я", "ђ", "ѕ", "і", "ј", "ў", "һ", "ԁ", "ԡ", "Ե", "Զ", "ա", "գ", "կ", "ն", "ս", "ր", "և", "֊", "ִ", "ָ", "ּ", "ׁ", "א", "ג", "ד", "ה", "ו", "ח", "י", "מ", "נ", "ס", "פ", "ץ", "צ", "ק", "ר", "ש", "ת", "،", "آ", "إ", "ئ", "ا", "ب", "ة", "ت", "ث", "ج", "ح", "خ", "د", "ذ", "ر", "ز", "ش", "ط", "ع", "ف", "ق", "ل", "م", "ن", "ه", "و", "ى", "ي", "ٹ", "پ", "چ", "ک", "گ", "ہ", "ی", "ރ", "ކ", "ं", "आ", "क", "ग", "च", "ज", "त", "न", "प", "फ", "म", "य", "र", "ल", "व", "श", "स", "ा", "ि", "ी", "ु", "ू", "े", "्", "চ", "জ", "ত", "ন", "প", "ব", "ম", "র", "্", "୨", "அ", "உ", "க", "த", "ய", "ர", "ல", "ள", "வ", "ா", "ி", "ு", "்", "ิ", "฿", "ไ", "་", "།", "ག", "ཆ", "ན", "པ", "ར", "ི", "ུ", "ེ", "ོ", "ដ", "ឋ", "ភ", "រ", "ស", "ា", "្", "᠇", "ᦁ", "ᦆ", "ᦇ", "ᦈ", "ᦗ", "ᦚ", "ḍ", "Ḥ", "ḥ", "Ḫ", "ḫ", "ḻ", "ṅ", "ṇ", "ṛ", "Ṣ", "ṣ", "Ṭ", "ṭ", "ẓ", "ạ", "ả", "ầ", "ẽ", "ệ", "ị", "Ỗ", "ỗ", "ớ", "ἔ", "Ἰ", "ῶ", " ", " ", " ", " ", " ", "​", "‌", "‍", "‎", "‏", "‐", "‑", "–", "—", "―", "‖", "‘", "’", "‚", "“", "”", "„", "†", "‡", "•", "…", "‬", " ", "′", "″", "‷", "‹", "›", "※", "‼", "⁄", "⁉", "⁠", "⁨", "⁴", "⁵", "⁸", "₂", "₤", "₩", "€", "₹", "℃", "№", "℗", "℠", "™", "⅄", "⅓", "⅗", "⅝", "Ⅰ", "Ⅱ", "ⅼ", "ⅽ", "←", "↑", "→", "↓", "↔", "↵", "⇚", "⇧", "∀", "∃", "∅", "∆", "∑", "−", "√", "≈", "≠", "≤", "≥", "⊞", "⌐", "⌥", "⏎", "①", "②", "③", "⒇", "Ⓟ", "─", "│", "├", "□", "►", "▽", "◄", "◆", "○", "◌", "●", "◟", "★", "☆", "☕", "☝", "☠", "☰", "☹", "☺", "♀", "♂", "♌", "♒", "♙", "♠", "♡", "♥", "♦", "♪", "♫", "⚜", "⚠", "⚡", "✅", "✈", "✊", "✋", "✌", "✓", "✔", "✨", "❌", "❤", "❶", "⟨", "⟩", "⭐", "ⲟ", "ⲡ", "ⲣ", "⽝", "、", "。", "《", "》", "「", "」", "『", "』", "【", "】", "〓", "〜", "〝", "〞", "〟", "〢", "あ", "い", "う", "お", "か", "が", "き", "げ", "こ", "ご", "さ", "ざ", "し", "じ", "す", "せ", "そ", "た", "ち", "っ", "て", "で", "と", "ど", "な", "に", "ね", "の", "は", "ま", "み", "め", "も", "や", "よ", "ら", "り", "わ", "を", "ん", "ゞ", "゠", "ァ", "ア", "イ", "オ", "ク", "コ", "シ", "セ", "ゼ", "ダ", "ツ", "ョ", "��", "ル", "レ", "ン", "・", "ー", "ㄟ", "㻳", "㻷", "䇂", "一", "丁", "七", "三", "不", "世", "中", "九", "五", "京", "伝", "传", "你", "例", "傲", "儿", "先", "典", "列", "動", "包", "千", "卣", "参", "叉", "反", "咖", "喱", "四", "国", "土", "地", "塗", "外", "大", "天", "太", "好", "妃", "学", "寫", "少", "尾", "山", "島", "帅", "年", "店", "庵", "庸", "强", "影", "往", "怖", "总", "恐", "息", "扁", "手", "投", "挂", "擾", "敏", "斯", "方", "施", "日", "昌", "显", "月", "期", "本", "東", "某", "树", "桂", "桐", "次", "民", "水", "洛", "津", "点", "煎", "燒", "爱", "猫", "理", "生", "电", "白", "百", "直", "真", "眼", "硫", "硼", "示", "祝", "神", "稿", "穆", "穴", "米", "紙", "經", "经", "胆", "膽", "苯", "英", "蒋", "衆", "西", "要", "覧", "览", "言", "記", "語", "诚", "谋", "貴", "起", "足", "辞", "运", "鄉", "酒", "醉", "释", "重", "金", "钠", "长", "閲", "阳", "院", "陳", "陽", "预", "騒", "黄", "Ꙑ", "ꙑ", "ꜣ", "교", "그", "기", "길", "낞", "니", "다", "당", "동", "로", "론", "릿", "면", "방", "백", "사", "삼", "선", "수", "습", "시", "식", "신", "실", "암", "용", "을", "인", "있", "준", "크", "토", "하", "확", "", "", "", "ff", "fi", "fl", "️", "﹩", "﹫", "ﻛ", "ﻜ", "ﻩ", "ﻴ", "", "!", "#", "$", "&", "'", "(", "+", ",", "-", "/", "3", ":", ">", "?", "A", "C", "D", "I", "R", "S", "T", "U", "Y", "_", "a", "b", "c", "d", "e", "f", "h", "i", "j", "k", "l", "m", "q", "r", "s", "t", "u", "v", "x", "y", "{", "|", "~", "「", "」", "、", "・", "ᄁ", "ᅭ", "¥", "", "�", "𐀏", "𐅇", "𐬀", "𐬌", "𐬎", "𐬑", "𐬙", "𐬞", "𐬭", "𐬱", "𔅴", "𔱰", "𖯗", "𖾺", "𗽋", "𝖆", "𝖊", "𝖍", "𝖑", "𝖓", "𝖘", "𝖙", "𝖜", "𞂾", "𞃳", "🇦", "🇨", "🇩", "🇪", "🇬", "🇭", "🇮", "🇱", "🇵", "🇷", "🇸", "🇺", "🇻", "🇾", "🇿", "🌌", "🌞", "🌟", "🌹", "🌺", "🍀", "🍁", "🍞", "🍾", "🎀", "🎈", "🎉", "🎊", "🎖", "🎶", "🎻", "🏅", "🏈", "🏚", "🏤", "🏥", "🏦", "🏻", "🏼", "🏽", "🏾", "🐌", "🐙", "🐚", "🐯", "🐱", "🐲", "👀", "👈", "👉", "👊", "👋", "👌", "👍", "👏", "👑", "👹", "👺", "💀", "💅", "💌", "💐", "💓", "💔", "💕", "💖", "💗", "💙", "💚", "💛", "💜", "💞", "💣", "💥", "💨", "💩", "💪", "💫", "💯", "💰", "📉", "📋", "📡", "📸", "🔕", "🔵", "🕶", "🖤", "🗞", "😀", "😁", "😂", "😃", "😄", "😅", "😆", "😈", "😉", "😊", "😋", "😌", "😍", "😎", "😏", "😐", "😑", "😒", "😓", "😔", "😕", "😖", "😘", "😛", "😜", "😝", "😞", "😟", "😡", "😢", "😤", "😥", "😧", "😨", "😩", "😪", "😫", "😬", "😭", "😮", "😯", "😰", "😱", "😲", "😳", "😵", "😶", "😹", "🙀", "🙂", "🙃", "🙄", "🙈", "🙌", "🙏", "🚨", "🚮", "🚽", "🛑", "🛒", "🛫", "🟩", "🤌", "🤍", "🤑", "🤓", "🤔", "🤗", "🤘", "🤛", "🤜", "🤝", "🤞", "🤡", "🤢", "🤣", "🤤", "🤦", "🤧", "🤨", "🤩", "🤫", "🤬", "🤭", "🤮", "🤯", "🤷", "🥋", "🥚", "🥰", "🥱", "🥲", "🥳", "🥵", "🥹", "🥺", "🦃", "🦜", "🦾", "🧊", "🧐", "🧠", "🩵", "🩷", "🩹", "🪙", "🪪", "🪲", "🪽", "🫀", "🫂", "🫠", "🫡", "🫢", "🫣", "🫶", "򟝥", "򟧷", "󉜹"]}
Free AI Image Generator No sign-up. Instant results. Open Now