initial commit

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,32 @@
+---
+library_name: granite_tsfm
+base_model: ibm-granite/granite-timeseries-ttm-r2
+tags:
+- ttm4hvac
+- tsfm
+- digital twin
+- hvac
+- energy
+- experiment
+license: apache-2.0
+datasets:
+- gft/ttm4hvac-target-default-train
+- gft/ttm4hvac-target-heat-test
+- gft/ttm4hvac-target-cool-test
+pipeline_tag: time-series-forecasting
+papers:
+- title: "Transfer learning of building dynamics digital twin for HVAC control with Time-series Foundation Model"
+  url: https://arxiv.org/abs/XXXX.XXXXX
+  authors: "Ferran Aran Domingo"
+---
+
+# TTM4HVAC – TinyTimeMixer for HVAC dynamics modeling 
+
+This repository contains the **TTM4HVAC – target-default** fine-tuned TinyTimeMixer model.
+
+It corresponds to the **“target-default” experiment** from the TTM4HVAC paper, where the model is fine-tuned solely on **default RBC data from the target building** (BestestAir, Denver climate).
+
+This checkpoint is part of the TTM4HVAC experiment family.  
+For the main model, recommended usage, detailed examples, and full documentation, please see:
+
+👉 **Main model card:** https://huggingface.co/gft/ttm4hvac

config.json

@@ -0,0 +1,74 @@
+{
+  "adaptive_patching_levels": 3,
+  "architectures": [
+    "TinyTimeMixerForPrediction"
+  ],
+  "categorical_vocab_size_list": null,
+  "context_length": 512,
+  "d_model": 192,
+  "d_model_scale": 3,
+  "decoder_adaptive_patching_levels": 0,
+  "decoder_d_model": 128,
+  "decoder_d_model_scale": 2,
+  "decoder_mode": "common_channel",
+  "decoder_num_layers": 2,
+  "decoder_raw_residual": false,
+  "distribution_output": "student_t",
+  "dropout": 0.4,
+  "enable_forecast_channel_mixing": true,
+  "exogenous_channel_indices": [
+    2,
+    3,
+    4,
+    5,
+    6,
+    7
+  ],
+  "expansion_factor": 2,
+  "fcm_context_length": 8,
+  "fcm_gated_attn": true,
+  "fcm_mix_layers": 2,
+  "fcm_prepend_past": true,
+  "fcm_prepend_past_offset": null,
+  "fcm_use_mixer": true,
+  "frequency_token_vocab_size": 10,
+  "gated_attn": true,
+  "head_dropout": 0.4,
+  "huber_delta": 1,
+  "init_embed": "pytorch",
+  "init_linear": "pytorch",
+  "init_processing": true,
+  "init_std": 0.02,
+  "loss": "mae",
+  "mask_value": 0,
+  "masked_context_length": null,
+  "mode": "common_channel",
+  "model_type": "tinytimemixer",
+  "norm_eps": 1e-05,
+  "norm_mlp": "LayerNorm",
+  "num_input_channels": 8,
+  "num_layers": 2,
+  "num_parallel_samples": 100,
+  "num_patches": 9,
+  "patch_last": true,
+  "patch_length": 64,
+  "patch_stride": 64,
+  "positional_encoding_type": "sincos",
+  "post_init": false,
+  "prediction_channel_indices": [
+    0,
+    1
+  ],
+  "prediction_filter_length": null,
+  "prediction_length": 96,
+  "quantile": 0.5,
+  "resolution_prefix_tuning": true,
+  "scaling": "std",
+  "self_attn": false,
+  "self_attn_heads": 1,
+  "stride_ratio": 1,
+  "torch_dtype": "float32",
+  "transformers_version": "4.55.0",
+  "use_decoder": true,
+  "use_positional_encoding": false
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2ceb635d77ad475bb9639b17f01d08761db8fbc4803eb20988b81cbf585277c9
+size 3798112

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:938f3270bd10f34b2dfa513993a50dd5d4777aa6f249de48a91536cd66ba9b04
+size 3184075

rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:216b8dd31569d03af0d2d439b084844dc4a82db0ad43c418dd1e4c83d412cc3a
+size 14645

scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd2606ce1efb8c07691aa91e4df6a3509b84b5b890a35928098943272a5c2fd2
+size 1465

trainer_state.json

@@ -0,0 +1,1603 @@
+{
+  "best_global_step": 208,
+  "best_metric": 0.305007666349411,
+  "best_model_checkpoint": "tmp/out/512-96-ft-l1-r2.1_common_channel_fcmCtx8_fcmLayers2_fcmChMixingTrue_stride24_bs512_lr0.0003_89b6/checkpoint-208",
+  "epoch": 104.0,
+  "eval_steps": 500,
+  "global_step": 208,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.6618412137031555,
+      "learning_rate": 0.0002999998149449555,
+      "loss": 0.3922,
+      "step": 2
+    },
+    {
+      "epoch": 1.0,
+      "eval_loss": 0.358045369386673,
+      "eval_runtime": 0.9433,
+      "eval_samples_per_second": 314.866,
+      "eval_steps_per_second": 1.06,
+      "step": 2
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 0.465237021446228,
+      "learning_rate": 0.0002999983345073394,
+      "loss": 0.3871,
+      "step": 4
+    },
+    {
+      "epoch": 2.0,
+      "eval_loss": 0.35514307022094727,
+      "eval_runtime": 0.984,
+      "eval_samples_per_second": 301.829,
+      "eval_steps_per_second": 1.016,
+      "step": 4
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 0.3141915500164032,
+      "learning_rate": 0.00029999537364671844,
+      "loss": 0.3814,
+      "step": 6
+    },
+    {
+      "epoch": 3.0,
+      "eval_loss": 0.3537431061267853,
+      "eval_runtime": 1.0649,
+      "eval_samples_per_second": 278.894,
+      "eval_steps_per_second": 0.939,
+      "step": 6
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 0.3491910696029663,
+      "learning_rate": 0.0002999909323923152,
+      "loss": 0.3733,
+      "step": 8
+    },
+    {
+      "epoch": 4.0,
+      "eval_loss": 0.3517496585845947,
+      "eval_runtime": 0.9441,
+      "eval_samples_per_second": 314.571,
+      "eval_steps_per_second": 1.059,
+      "step": 8
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 0.29127219319343567,
+      "learning_rate": 0.000299985010787963,
+      "loss": 0.3749,
+      "step": 10
+    },
+    {
+      "epoch": 5.0,
+      "eval_loss": 0.35048815608024597,
+      "eval_runtime": 0.9185,
+      "eval_samples_per_second": 323.353,
+      "eval_steps_per_second": 1.089,
+      "step": 10
+    },
+    {
+      "epoch": 6.0,
+      "grad_norm": 0.33912065625190735,
+      "learning_rate": 0.0002999776088921057,
+      "loss": 0.3708,
+      "step": 12
+    },
+    {
+      "epoch": 6.0,
+      "eval_loss": 0.35033392906188965,
+      "eval_runtime": 0.9493,
+      "eval_samples_per_second": 312.868,
+      "eval_steps_per_second": 1.053,
+      "step": 12
+    },
+    {
+      "epoch": 7.0,
+      "grad_norm": 0.311758428812027,
+      "learning_rate": 0.0002999687267777971,
+      "loss": 0.3685,
+      "step": 14
+    },
+    {
+      "epoch": 7.0,
+      "eval_loss": 0.35117536783218384,
+      "eval_runtime": 0.7975,
+      "eval_samples_per_second": 372.426,
+      "eval_steps_per_second": 1.254,
+      "step": 14
+    },
+    {
+      "epoch": 8.0,
+      "grad_norm": 0.30109912157058716,
+      "learning_rate": 0.0002999583645327,
+      "loss": 0.3639,
+      "step": 16
+    },
+    {
+      "epoch": 8.0,
+      "eval_loss": 0.35066911578178406,
+      "eval_runtime": 0.7655,
+      "eval_samples_per_second": 388.003,
+      "eval_steps_per_second": 1.306,
+      "step": 16
+    },
+    {
+      "epoch": 9.0,
+      "grad_norm": 0.3332655131816864,
+      "learning_rate": 0.0002999465222590856,
+      "loss": 0.3655,
+      "step": 18
+    },
+    {
+      "epoch": 9.0,
+      "eval_loss": 0.350115031003952,
+      "eval_runtime": 0.7681,
+      "eval_samples_per_second": 386.655,
+      "eval_steps_per_second": 1.302,
+      "step": 18
+    },
+    {
+      "epoch": 10.0,
+      "grad_norm": 0.36181315779685974,
+      "learning_rate": 0.00029993320007383234,
+      "loss": 0.3627,
+      "step": 20
+    },
+    {
+      "epoch": 10.0,
+      "eval_loss": 0.35086938738822937,
+      "eval_runtime": 0.7303,
+      "eval_samples_per_second": 406.705,
+      "eval_steps_per_second": 1.369,
+      "step": 20
+    },
+    {
+      "epoch": 11.0,
+      "grad_norm": 0.3522287607192993,
+      "learning_rate": 0.0002999183981084249,
+      "loss": 0.3575,
+      "step": 22
+    },
+    {
+      "epoch": 11.0,
+      "eval_loss": 0.3501550853252411,
+      "eval_runtime": 0.6638,
+      "eval_samples_per_second": 447.418,
+      "eval_steps_per_second": 1.506,
+      "step": 22
+    },
+    {
+      "epoch": 12.0,
+      "grad_norm": 0.48265793919563293,
+      "learning_rate": 0.0002999021165089526,
+      "loss": 0.356,
+      "step": 24
+    },
+    {
+      "epoch": 12.0,
+      "eval_loss": 0.34953632950782776,
+      "eval_runtime": 0.6847,
+      "eval_samples_per_second": 433.784,
+      "eval_steps_per_second": 1.461,
+      "step": 24
+    },
+    {
+      "epoch": 13.0,
+      "grad_norm": 0.452486515045166,
+      "learning_rate": 0.0002998843554361083,
+      "loss": 0.3572,
+      "step": 26
+    },
+    {
+      "epoch": 13.0,
+      "eval_loss": 0.34998443722724915,
+      "eval_runtime": 0.6957,
+      "eval_samples_per_second": 426.915,
+      "eval_steps_per_second": 1.437,
+      "step": 26
+    },
+    {
+      "epoch": 14.0,
+      "grad_norm": 0.5479589700698853,
+      "learning_rate": 0.0002998651150651866,
+      "loss": 0.3602,
+      "step": 28
+    },
+    {
+      "epoch": 14.0,
+      "eval_loss": 0.3498837649822235,
+      "eval_runtime": 0.7349,
+      "eval_samples_per_second": 404.119,
+      "eval_steps_per_second": 1.361,
+      "step": 28
+    },
+    {
+      "epoch": 15.0,
+      "grad_norm": 0.43432193994522095,
+      "learning_rate": 0.00029984439558608224,
+      "loss": 0.3552,
+      "step": 30
+    },
+    {
+      "epoch": 15.0,
+      "eval_loss": 0.34781429171562195,
+      "eval_runtime": 0.6365,
+      "eval_samples_per_second": 466.635,
+      "eval_steps_per_second": 1.571,
+      "step": 30
+    },
+    {
+      "epoch": 16.0,
+      "grad_norm": 0.35709846019744873,
+      "learning_rate": 0.00029982219720328803,
+      "loss": 0.3442,
+      "step": 32
+    },
+    {
+      "epoch": 16.0,
+      "eval_loss": 0.34714677929878235,
+      "eval_runtime": 0.7389,
+      "eval_samples_per_second": 401.956,
+      "eval_steps_per_second": 1.353,
+      "step": 32
+    },
+    {
+      "epoch": 17.0,
+      "grad_norm": 0.3510877192020416,
+      "learning_rate": 0.00029979852013589306,
+      "loss": 0.3466,
+      "step": 34
+    },
+    {
+      "epoch": 17.0,
+      "eval_loss": 0.3476266860961914,
+      "eval_runtime": 0.582,
+      "eval_samples_per_second": 510.287,
+      "eval_steps_per_second": 1.718,
+      "step": 34
+    },
+    {
+      "epoch": 18.0,
+      "grad_norm": 0.6324598789215088,
+      "learning_rate": 0.0002997733646175805,
+      "loss": 0.3495,
+      "step": 36
+    },
+    {
+      "epoch": 18.0,
+      "eval_loss": 0.34712979197502136,
+      "eval_runtime": 0.6306,
+      "eval_samples_per_second": 470.991,
+      "eval_steps_per_second": 1.586,
+      "step": 36
+    },
+    {
+      "epoch": 19.0,
+      "grad_norm": 0.3290903866291046,
+      "learning_rate": 0.00029974673089662506,
+      "loss": 0.343,
+      "step": 38
+    },
+    {
+      "epoch": 19.0,
+      "eval_loss": 0.3468128442764282,
+      "eval_runtime": 0.6393,
+      "eval_samples_per_second": 464.534,
+      "eval_steps_per_second": 1.564,
+      "step": 38
+    },
+    {
+      "epoch": 20.0,
+      "grad_norm": 0.3430045545101166,
+      "learning_rate": 0.00029971861923589095,
+      "loss": 0.342,
+      "step": 40
+    },
+    {
+      "epoch": 20.0,
+      "eval_loss": 0.3459309935569763,
+      "eval_runtime": 0.7921,
+      "eval_samples_per_second": 374.931,
+      "eval_steps_per_second": 1.262,
+      "step": 40
+    },
+    {
+      "epoch": 21.0,
+      "grad_norm": 0.39738816022872925,
+      "learning_rate": 0.0002996890299128288,
+      "loss": 0.3444,
+      "step": 42
+    },
+    {
+      "epoch": 21.0,
+      "eval_loss": 0.3452853560447693,
+      "eval_runtime": 0.7918,
+      "eval_samples_per_second": 375.108,
+      "eval_steps_per_second": 1.263,
+      "step": 42
+    },
+    {
+      "epoch": 22.0,
+      "grad_norm": 0.3685533106327057,
+      "learning_rate": 0.00029965796321947336,
+      "loss": 0.3368,
+      "step": 44
+    },
+    {
+      "epoch": 22.0,
+      "eval_loss": 0.34537574648857117,
+      "eval_runtime": 0.715,
+      "eval_samples_per_second": 415.38,
+      "eval_steps_per_second": 1.399,
+      "step": 44
+    },
+    {
+      "epoch": 23.0,
+      "grad_norm": 0.41565972566604614,
+      "learning_rate": 0.00029962541946244024,
+      "loss": 0.3337,
+      "step": 46
+    },
+    {
+      "epoch": 23.0,
+      "eval_loss": 0.34441086649894714,
+      "eval_runtime": 0.5609,
+      "eval_samples_per_second": 529.532,
+      "eval_steps_per_second": 1.783,
+      "step": 46
+    },
+    {
+      "epoch": 24.0,
+      "grad_norm": 0.36327725648880005,
+      "learning_rate": 0.00029959139896292323,
+      "loss": 0.3342,
+      "step": 48
+    },
+    {
+      "epoch": 24.0,
+      "eval_loss": 0.34298038482666016,
+      "eval_runtime": 0.672,
+      "eval_samples_per_second": 441.965,
+      "eval_steps_per_second": 1.488,
+      "step": 48
+    },
+    {
+      "epoch": 25.0,
+      "grad_norm": 0.4043494164943695,
+      "learning_rate": 0.0002995559020566911,
+      "loss": 0.3414,
+      "step": 50
+    },
+    {
+      "epoch": 25.0,
+      "eval_loss": 0.3428707420825958,
+      "eval_runtime": 0.6633,
+      "eval_samples_per_second": 447.772,
+      "eval_steps_per_second": 1.508,
+      "step": 50
+    },
+    {
+      "epoch": 26.0,
+      "grad_norm": 0.40012046694755554,
+      "learning_rate": 0.0002995189290940837,
+      "loss": 0.3387,
+      "step": 52
+    },
+    {
+      "epoch": 26.0,
+      "eval_loss": 0.3441244959831238,
+      "eval_runtime": 0.7445,
+      "eval_samples_per_second": 398.927,
+      "eval_steps_per_second": 1.343,
+      "step": 52
+    },
+    {
+      "epoch": 27.0,
+      "grad_norm": 0.4684393107891083,
+      "learning_rate": 0.0002994804804400094,
+      "loss": 0.3343,
+      "step": 54
+    },
+    {
+      "epoch": 27.0,
+      "eval_loss": 0.3433794677257538,
+      "eval_runtime": 0.681,
+      "eval_samples_per_second": 436.153,
+      "eval_steps_per_second": 1.469,
+      "step": 54
+    },
+    {
+      "epoch": 28.0,
+      "grad_norm": 0.4722350239753723,
+      "learning_rate": 0.00029944055647394087,
+      "loss": 0.3366,
+      "step": 56
+    },
+    {
+      "epoch": 28.0,
+      "eval_loss": 0.3417203724384308,
+      "eval_runtime": 0.6481,
+      "eval_samples_per_second": 458.289,
+      "eval_steps_per_second": 1.543,
+      "step": 56
+    },
+    {
+      "epoch": 29.0,
+      "grad_norm": 0.5819035768508911,
+      "learning_rate": 0.0002993991575899116,
+      "loss": 0.3338,
+      "step": 58
+    },
+    {
+      "epoch": 29.0,
+      "eval_loss": 0.3421356976032257,
+      "eval_runtime": 0.8299,
+      "eval_samples_per_second": 357.891,
+      "eval_steps_per_second": 1.205,
+      "step": 58
+    },
+    {
+      "epoch": 30.0,
+      "grad_norm": 0.4985131621360779,
+      "learning_rate": 0.0002993562841965118,
+      "loss": 0.3314,
+      "step": 60
+    },
+    {
+      "epoch": 30.0,
+      "eval_loss": 0.34163638949394226,
+      "eval_runtime": 0.6224,
+      "eval_samples_per_second": 477.149,
+      "eval_steps_per_second": 1.607,
+      "step": 60
+    },
+    {
+      "epoch": 31.0,
+      "grad_norm": 0.5123302340507507,
+      "learning_rate": 0.00029931193671688453,
+      "loss": 0.33,
+      "step": 62
+    },
+    {
+      "epoch": 31.0,
+      "eval_loss": 0.33960819244384766,
+      "eval_runtime": 0.7702,
+      "eval_samples_per_second": 385.59,
+      "eval_steps_per_second": 1.298,
+      "step": 62
+    },
+    {
+      "epoch": 32.0,
+      "grad_norm": 0.34553447365760803,
+      "learning_rate": 0.0002992661155887215,
+      "loss": 0.3253,
+      "step": 64
+    },
+    {
+      "epoch": 32.0,
+      "eval_loss": 0.33889296650886536,
+      "eval_runtime": 0.6787,
+      "eval_samples_per_second": 437.601,
+      "eval_steps_per_second": 1.473,
+      "step": 64
+    },
+    {
+      "epoch": 33.0,
+      "grad_norm": 0.4334942698478699,
+      "learning_rate": 0.0002992188212642587,
+      "loss": 0.3221,
+      "step": 66
+    },
+    {
+      "epoch": 33.0,
+      "eval_loss": 0.3390357494354248,
+      "eval_runtime": 0.7883,
+      "eval_samples_per_second": 376.773,
+      "eval_steps_per_second": 1.269,
+      "step": 66
+    },
+    {
+      "epoch": 34.0,
+      "grad_norm": 0.5000684261322021,
+      "learning_rate": 0.0002991700542102722,
+      "loss": 0.3228,
+      "step": 68
+    },
+    {
+      "epoch": 34.0,
+      "eval_loss": 0.3387429118156433,
+      "eval_runtime": 0.7066,
+      "eval_samples_per_second": 420.345,
+      "eval_steps_per_second": 1.415,
+      "step": 68
+    },
+    {
+      "epoch": 35.0,
+      "grad_norm": 0.5403483510017395,
+      "learning_rate": 0.000299119814908073,
+      "loss": 0.3246,
+      "step": 70
+    },
+    {
+      "epoch": 35.0,
+      "eval_loss": 0.337623655796051,
+      "eval_runtime": 0.5666,
+      "eval_samples_per_second": 524.176,
+      "eval_steps_per_second": 1.765,
+      "step": 70
+    },
+    {
+      "epoch": 36.0,
+      "grad_norm": 0.3801553547382355,
+      "learning_rate": 0.00029906810385350283,
+      "loss": 0.3215,
+      "step": 72
+    },
+    {
+      "epoch": 36.0,
+      "eval_loss": 0.3370514214038849,
+      "eval_runtime": 0.7169,
+      "eval_samples_per_second": 414.265,
+      "eval_steps_per_second": 1.395,
+      "step": 72
+    },
+    {
+      "epoch": 37.0,
+      "grad_norm": 0.6648715138435364,
+      "learning_rate": 0.00029901492155692876,
+      "loss": 0.3205,
+      "step": 74
+    },
+    {
+      "epoch": 37.0,
+      "eval_loss": 0.3367176651954651,
+      "eval_runtime": 0.6219,
+      "eval_samples_per_second": 477.599,
+      "eval_steps_per_second": 1.608,
+      "step": 74
+    },
+    {
+      "epoch": 38.0,
+      "grad_norm": 0.38467201590538025,
+      "learning_rate": 0.0002989602685432388,
+      "loss": 0.3175,
+      "step": 76
+    },
+    {
+      "epoch": 38.0,
+      "eval_loss": 0.3385818302631378,
+      "eval_runtime": 0.7477,
+      "eval_samples_per_second": 397.199,
+      "eval_steps_per_second": 1.337,
+      "step": 76
+    },
+    {
+      "epoch": 39.0,
+      "grad_norm": 0.5086773633956909,
+      "learning_rate": 0.00029890414535183583,
+      "loss": 0.3188,
+      "step": 78
+    },
+    {
+      "epoch": 39.0,
+      "eval_loss": 0.33492594957351685,
+      "eval_runtime": 0.7572,
+      "eval_samples_per_second": 392.254,
+      "eval_steps_per_second": 1.321,
+      "step": 78
+    },
+    {
+      "epoch": 40.0,
+      "grad_norm": 0.4003610610961914,
+      "learning_rate": 0.00029884655253663344,
+      "loss": 0.3197,
+      "step": 80
+    },
+    {
+      "epoch": 40.0,
+      "eval_loss": 0.33226659893989563,
+      "eval_runtime": 0.738,
+      "eval_samples_per_second": 402.452,
+      "eval_steps_per_second": 1.355,
+      "step": 80
+    },
+    {
+      "epoch": 41.0,
+      "grad_norm": 0.4991908371448517,
+      "learning_rate": 0.00029878749066604925,
+      "loss": 0.313,
+      "step": 82
+    },
+    {
+      "epoch": 41.0,
+      "eval_loss": 0.331875741481781,
+      "eval_runtime": 0.7609,
+      "eval_samples_per_second": 390.343,
+      "eval_steps_per_second": 1.314,
+      "step": 82
+    },
+    {
+      "epoch": 42.0,
+      "grad_norm": 0.450600802898407,
+      "learning_rate": 0.0002987269603230001,
+      "loss": 0.3142,
+      "step": 84
+    },
+    {
+      "epoch": 42.0,
+      "eval_loss": 0.33316686749458313,
+      "eval_runtime": 0.6259,
+      "eval_samples_per_second": 474.552,
+      "eval_steps_per_second": 1.598,
+      "step": 84
+    },
+    {
+      "epoch": 43.0,
+      "grad_norm": 0.7004669308662415,
+      "learning_rate": 0.00029866496210489605,
+      "loss": 0.3112,
+      "step": 86
+    },
+    {
+      "epoch": 43.0,
+      "eval_loss": 0.3317732810974121,
+      "eval_runtime": 0.7352,
+      "eval_samples_per_second": 403.996,
+      "eval_steps_per_second": 1.36,
+      "step": 86
+    },
+    {
+      "epoch": 44.0,
+      "grad_norm": 0.4675546884536743,
+      "learning_rate": 0.0002986014966236345,
+      "loss": 0.3118,
+      "step": 88
+    },
+    {
+      "epoch": 44.0,
+      "eval_loss": 0.32955238223075867,
+      "eval_runtime": 0.8122,
+      "eval_samples_per_second": 365.683,
+      "eval_steps_per_second": 1.231,
+      "step": 88
+    },
+    {
+      "epoch": 45.0,
+      "grad_norm": 0.5740877389907837,
+      "learning_rate": 0.00029853656450559414,
+      "loss": 0.3083,
+      "step": 90
+    },
+    {
+      "epoch": 45.0,
+      "eval_loss": 0.32967710494995117,
+      "eval_runtime": 0.7609,
+      "eval_samples_per_second": 390.346,
+      "eval_steps_per_second": 1.314,
+      "step": 90
+    },
+    {
+      "epoch": 46.0,
+      "grad_norm": 0.5421927571296692,
+      "learning_rate": 0.00029847016639162867,
+      "loss": 0.3125,
+      "step": 92
+    },
+    {
+      "epoch": 46.0,
+      "eval_loss": 0.33168312907218933,
+      "eval_runtime": 0.6488,
+      "eval_samples_per_second": 457.788,
+      "eval_steps_per_second": 1.541,
+      "step": 92
+    },
+    {
+      "epoch": 47.0,
+      "grad_norm": 0.6857365369796753,
+      "learning_rate": 0.0002984023029370609,
+      "loss": 0.3035,
+      "step": 94
+    },
+    {
+      "epoch": 47.0,
+      "eval_loss": 0.3298337161540985,
+      "eval_runtime": 0.6488,
+      "eval_samples_per_second": 457.751,
+      "eval_steps_per_second": 1.541,
+      "step": 94
+    },
+    {
+      "epoch": 48.0,
+      "grad_norm": 0.8388419151306152,
+      "learning_rate": 0.0002983329748116755,
+      "loss": 0.3031,
+      "step": 96
+    },
+    {
+      "epoch": 48.0,
+      "eval_loss": 0.32884910702705383,
+      "eval_runtime": 0.6776,
+      "eval_samples_per_second": 438.283,
+      "eval_steps_per_second": 1.476,
+      "step": 96
+    },
+    {
+      "epoch": 49.0,
+      "grad_norm": 0.46255478262901306,
+      "learning_rate": 0.00029826218269971314,
+      "loss": 0.305,
+      "step": 98
+    },
+    {
+      "epoch": 49.0,
+      "eval_loss": 0.3276749551296234,
+      "eval_runtime": 0.733,
+      "eval_samples_per_second": 405.188,
+      "eval_steps_per_second": 1.364,
+      "step": 98
+    },
+    {
+      "epoch": 50.0,
+      "grad_norm": 0.44940125942230225,
+      "learning_rate": 0.0002981899272998634,
+      "loss": 0.302,
+      "step": 100
+    },
+    {
+      "epoch": 50.0,
+      "eval_loss": 0.3282870352268219,
+      "eval_runtime": 0.7162,
+      "eval_samples_per_second": 414.71,
+      "eval_steps_per_second": 1.396,
+      "step": 100
+    },
+    {
+      "epoch": 51.0,
+      "grad_norm": 0.5168047547340393,
+      "learning_rate": 0.0002981162093252579,
+      "loss": 0.2989,
+      "step": 102
+    },
+    {
+      "epoch": 51.0,
+      "eval_loss": 0.33124735951423645,
+      "eval_runtime": 0.629,
+      "eval_samples_per_second": 472.143,
+      "eval_steps_per_second": 1.59,
+      "step": 102
+    },
+    {
+      "epoch": 52.0,
+      "grad_norm": 0.5312138795852661,
+      "learning_rate": 0.00029804102950346334,
+      "loss": 0.2957,
+      "step": 104
+    },
+    {
+      "epoch": 52.0,
+      "eval_loss": 0.3266555964946747,
+      "eval_runtime": 0.6431,
+      "eval_samples_per_second": 461.793,
+      "eval_steps_per_second": 1.555,
+      "step": 104
+    },
+    {
+      "epoch": 53.0,
+      "grad_norm": 0.5349692702293396,
+      "learning_rate": 0.0002979643885764741,
+      "loss": 0.2975,
+      "step": 106
+    },
+    {
+      "epoch": 53.0,
+      "eval_loss": 0.3233616352081299,
+      "eval_runtime": 0.8057,
+      "eval_samples_per_second": 368.612,
+      "eval_steps_per_second": 1.241,
+      "step": 106
+    },
+    {
+      "epoch": 54.0,
+      "grad_norm": 0.36133870482444763,
+      "learning_rate": 0.00029788628730070533,
+      "loss": 0.2957,
+      "step": 108
+    },
+    {
+      "epoch": 54.0,
+      "eval_loss": 0.323390930891037,
+      "eval_runtime": 0.6815,
+      "eval_samples_per_second": 435.829,
+      "eval_steps_per_second": 1.467,
+      "step": 108
+    },
+    {
+      "epoch": 55.0,
+      "grad_norm": 0.4435558021068573,
+      "learning_rate": 0.00029780672644698494,
+      "loss": 0.2956,
+      "step": 110
+    },
+    {
+      "epoch": 55.0,
+      "eval_loss": 0.3230491578578949,
+      "eval_runtime": 0.6254,
+      "eval_samples_per_second": 474.93,
+      "eval_steps_per_second": 1.599,
+      "step": 110
+    },
+    {
+      "epoch": 56.0,
+      "grad_norm": 0.4963320791721344,
+      "learning_rate": 0.0002977257068005465,
+      "loss": 0.2927,
+      "step": 112
+    },
+    {
+      "epoch": 56.0,
+      "eval_loss": 0.321744829416275,
+      "eval_runtime": 0.7031,
+      "eval_samples_per_second": 422.43,
+      "eval_steps_per_second": 1.422,
+      "step": 112
+    },
+    {
+      "epoch": 57.0,
+      "grad_norm": 0.47572246193885803,
+      "learning_rate": 0.0002976432291610213,
+      "loss": 0.2945,
+      "step": 114
+    },
+    {
+      "epoch": 57.0,
+      "eval_loss": 0.32216939330101013,
+      "eval_runtime": 0.77,
+      "eval_samples_per_second": 385.727,
+      "eval_steps_per_second": 1.299,
+      "step": 114
+    },
+    {
+      "epoch": 58.0,
+      "grad_norm": 0.4532373547554016,
+      "learning_rate": 0.00029755929434243023,
+      "loss": 0.2912,
+      "step": 116
+    },
+    {
+      "epoch": 58.0,
+      "eval_loss": 0.32353639602661133,
+      "eval_runtime": 0.7486,
+      "eval_samples_per_second": 396.744,
+      "eval_steps_per_second": 1.336,
+      "step": 116
+    },
+    {
+      "epoch": 59.0,
+      "grad_norm": 0.37667471170425415,
+      "learning_rate": 0.00029747390317317603,
+      "loss": 0.2892,
+      "step": 118
+    },
+    {
+      "epoch": 59.0,
+      "eval_loss": 0.3225230276584625,
+      "eval_runtime": 0.6999,
+      "eval_samples_per_second": 424.335,
+      "eval_steps_per_second": 1.429,
+      "step": 118
+    },
+    {
+      "epoch": 60.0,
+      "grad_norm": 0.48530539870262146,
+      "learning_rate": 0.0002973870564960352,
+      "loss": 0.2874,
+      "step": 120
+    },
+    {
+      "epoch": 60.0,
+      "eval_loss": 0.32104259729385376,
+      "eval_runtime": 0.6522,
+      "eval_samples_per_second": 455.417,
+      "eval_steps_per_second": 1.533,
+      "step": 120
+    },
+    {
+      "epoch": 61.0,
+      "grad_norm": 0.46407485008239746,
+      "learning_rate": 0.00029729875516814935,
+      "loss": 0.2836,
+      "step": 122
+    },
+    {
+      "epoch": 61.0,
+      "eval_loss": 0.32013222575187683,
+      "eval_runtime": 0.8233,
+      "eval_samples_per_second": 360.757,
+      "eval_steps_per_second": 1.215,
+      "step": 122
+    },
+    {
+      "epoch": 62.0,
+      "grad_norm": 0.6122850179672241,
+      "learning_rate": 0.0002972090000610169,
+      "loss": 0.2894,
+      "step": 124
+    },
+    {
+      "epoch": 62.0,
+      "eval_loss": 0.3198564350605011,
+      "eval_runtime": 0.6743,
+      "eval_samples_per_second": 440.478,
+      "eval_steps_per_second": 1.483,
+      "step": 124
+    },
+    {
+      "epoch": 63.0,
+      "grad_norm": 0.5037839412689209,
+      "learning_rate": 0.0002971177920604845,
+      "loss": 0.2891,
+      "step": 126
+    },
+    {
+      "epoch": 63.0,
+      "eval_loss": 0.31861403584480286,
+      "eval_runtime": 0.6524,
+      "eval_samples_per_second": 455.235,
+      "eval_steps_per_second": 1.533,
+      "step": 126
+    },
+    {
+      "epoch": 64.0,
+      "grad_norm": 0.4292280972003937,
+      "learning_rate": 0.00029702513206673827,
+      "loss": 0.2895,
+      "step": 128
+    },
+    {
+      "epoch": 64.0,
+      "eval_loss": 0.32153990864753723,
+      "eval_runtime": 0.6418,
+      "eval_samples_per_second": 462.761,
+      "eval_steps_per_second": 1.558,
+      "step": 128
+    },
+    {
+      "epoch": 65.0,
+      "grad_norm": 0.4504665732383728,
+      "learning_rate": 0.000296931020994295,
+      "loss": 0.2829,
+      "step": 130
+    },
+    {
+      "epoch": 65.0,
+      "eval_loss": 0.3232710063457489,
+      "eval_runtime": 0.7745,
+      "eval_samples_per_second": 383.487,
+      "eval_steps_per_second": 1.291,
+      "step": 130
+    },
+    {
+      "epoch": 66.0,
+      "grad_norm": 0.6219745874404907,
+      "learning_rate": 0.000296835459771993,
+      "loss": 0.2859,
+      "step": 132
+    },
+    {
+      "epoch": 66.0,
+      "eval_loss": 0.3181236982345581,
+      "eval_runtime": 0.7429,
+      "eval_samples_per_second": 399.762,
+      "eval_steps_per_second": 1.346,
+      "step": 132
+    },
+    {
+      "epoch": 67.0,
+      "grad_norm": 0.37324589490890503,
+      "learning_rate": 0.0002967384493429829,
+      "loss": 0.2798,
+      "step": 134
+    },
+    {
+      "epoch": 67.0,
+      "eval_loss": 0.31664231419563293,
+      "eval_runtime": 0.6029,
+      "eval_samples_per_second": 492.583,
+      "eval_steps_per_second": 1.659,
+      "step": 134
+    },
+    {
+      "epoch": 68.0,
+      "grad_norm": 0.48705121874809265,
+      "learning_rate": 0.0002966399906647185,
+      "loss": 0.281,
+      "step": 136
+    },
+    {
+      "epoch": 68.0,
+      "eval_loss": 0.3166642189025879,
+      "eval_runtime": 0.7131,
+      "eval_samples_per_second": 416.468,
+      "eval_steps_per_second": 1.402,
+      "step": 136
+    },
+    {
+      "epoch": 69.0,
+      "grad_norm": 0.4066949188709259,
+      "learning_rate": 0.0002965400847089472,
+      "loss": 0.2822,
+      "step": 138
+    },
+    {
+      "epoch": 69.0,
+      "eval_loss": 0.3151547312736511,
+      "eval_runtime": 0.7062,
+      "eval_samples_per_second": 420.548,
+      "eval_steps_per_second": 1.416,
+      "step": 138
+    },
+    {
+      "epoch": 70.0,
+      "grad_norm": 0.3984168469905853,
+      "learning_rate": 0.00029643873246170045,
+      "loss": 0.2801,
+      "step": 140
+    },
+    {
+      "epoch": 70.0,
+      "eval_loss": 0.314954549074173,
+      "eval_runtime": 0.6916,
+      "eval_samples_per_second": 429.454,
+      "eval_steps_per_second": 1.446,
+      "step": 140
+    },
+    {
+      "epoch": 71.0,
+      "grad_norm": 0.4139808118343353,
+      "learning_rate": 0.000296335934923284,
+      "loss": 0.2815,
+      "step": 142
+    },
+    {
+      "epoch": 71.0,
+      "eval_loss": 0.31498512625694275,
+      "eval_runtime": 0.6363,
+      "eval_samples_per_second": 466.781,
+      "eval_steps_per_second": 1.572,
+      "step": 142
+    },
+    {
+      "epoch": 72.0,
+      "grad_norm": 0.4868161678314209,
+      "learning_rate": 0.0002962316931082681,
+      "loss": 0.2781,
+      "step": 144
+    },
+    {
+      "epoch": 72.0,
+      "eval_loss": 0.31632447242736816,
+      "eval_runtime": 0.7076,
+      "eval_samples_per_second": 419.738,
+      "eval_steps_per_second": 1.413,
+      "step": 144
+    },
+    {
+      "epoch": 73.0,
+      "grad_norm": 0.41822776198387146,
+      "learning_rate": 0.0002961260080454773,
+      "loss": 0.2771,
+      "step": 146
+    },
+    {
+      "epoch": 73.0,
+      "eval_loss": 0.31577062606811523,
+      "eval_runtime": 0.6483,
+      "eval_samples_per_second": 458.136,
+      "eval_steps_per_second": 1.543,
+      "step": 146
+    },
+    {
+      "epoch": 74.0,
+      "grad_norm": 0.5603657960891724,
+      "learning_rate": 0.0002960188807779805,
+      "loss": 0.2732,
+      "step": 148
+    },
+    {
+      "epoch": 74.0,
+      "eval_loss": 0.31435203552246094,
+      "eval_runtime": 0.691,
+      "eval_samples_per_second": 429.801,
+      "eval_steps_per_second": 1.447,
+      "step": 148
+    },
+    {
+      "epoch": 75.0,
+      "grad_norm": 0.5573254227638245,
+      "learning_rate": 0.0002959103123630807,
+      "loss": 0.2766,
+      "step": 150
+    },
+    {
+      "epoch": 75.0,
+      "eval_loss": 0.3145710527896881,
+      "eval_runtime": 0.5527,
+      "eval_samples_per_second": 537.332,
+      "eval_steps_per_second": 1.809,
+      "step": 150
+    },
+    {
+      "epoch": 76.0,
+      "grad_norm": 0.4642890393733978,
+      "learning_rate": 0.0002958003038723042,
+      "loss": 0.2798,
+      "step": 152
+    },
+    {
+      "epoch": 76.0,
+      "eval_loss": 0.3158363401889801,
+      "eval_runtime": 0.6733,
+      "eval_samples_per_second": 441.12,
+      "eval_steps_per_second": 1.485,
+      "step": 152
+    },
+    {
+      "epoch": 77.0,
+      "grad_norm": 0.4938775300979614,
+      "learning_rate": 0.00029568885639139053,
+      "loss": 0.2753,
+      "step": 154
+    },
+    {
+      "epoch": 77.0,
+      "eval_loss": 0.31477540731430054,
+      "eval_runtime": 0.7417,
+      "eval_samples_per_second": 400.422,
+      "eval_steps_per_second": 1.348,
+      "step": 154
+    },
+    {
+      "epoch": 78.0,
+      "grad_norm": 0.39144381880760193,
+      "learning_rate": 0.00029557597102028123,
+      "loss": 0.2732,
+      "step": 156
+    },
+    {
+      "epoch": 78.0,
+      "eval_loss": 0.31341299414634705,
+      "eval_runtime": 0.7854,
+      "eval_samples_per_second": 378.137,
+      "eval_steps_per_second": 1.273,
+      "step": 156
+    },
+    {
+      "epoch": 79.0,
+      "grad_norm": 0.4929453432559967,
+      "learning_rate": 0.00029546164887310933,
+      "loss": 0.2711,
+      "step": 158
+    },
+    {
+      "epoch": 79.0,
+      "eval_loss": 0.31385812163352966,
+      "eval_runtime": 0.6657,
+      "eval_samples_per_second": 446.114,
+      "eval_steps_per_second": 1.502,
+      "step": 158
+    },
+    {
+      "epoch": 80.0,
+      "grad_norm": 0.40074649453163147,
+      "learning_rate": 0.0002953458910781883,
+      "loss": 0.2677,
+      "step": 160
+    },
+    {
+      "epoch": 80.0,
+      "eval_loss": 0.3135402500629425,
+      "eval_runtime": 0.6596,
+      "eval_samples_per_second": 450.303,
+      "eval_steps_per_second": 1.516,
+      "step": 160
+    },
+    {
+      "epoch": 81.0,
+      "grad_norm": 0.41622215509414673,
+      "learning_rate": 0.0002952286987780008,
+      "loss": 0.2668,
+      "step": 162
+    },
+    {
+      "epoch": 81.0,
+      "eval_loss": 0.31341665983200073,
+      "eval_runtime": 0.5644,
+      "eval_samples_per_second": 526.236,
+      "eval_steps_per_second": 1.772,
+      "step": 162
+    },
+    {
+      "epoch": 82.0,
+      "grad_norm": 0.5020734667778015,
+      "learning_rate": 0.0002951100731291876,
+      "loss": 0.2734,
+      "step": 164
+    },
+    {
+      "epoch": 82.0,
+      "eval_loss": 0.3154440224170685,
+      "eval_runtime": 0.7427,
+      "eval_samples_per_second": 399.916,
+      "eval_steps_per_second": 1.347,
+      "step": 164
+    },
+    {
+      "epoch": 83.0,
+      "grad_norm": 0.7120861411094666,
+      "learning_rate": 0.0002949900153025359,
+      "loss": 0.2712,
+      "step": 166
+    },
+    {
+      "epoch": 83.0,
+      "eval_loss": 0.3155328929424286,
+      "eval_runtime": 0.734,
+      "eval_samples_per_second": 404.611,
+      "eval_steps_per_second": 1.362,
+      "step": 166
+    },
+    {
+      "epoch": 84.0,
+      "grad_norm": 0.5501344799995422,
+      "learning_rate": 0.00029486852648296806,
+      "loss": 0.2705,
+      "step": 168
+    },
+    {
+      "epoch": 84.0,
+      "eval_loss": 0.31369757652282715,
+      "eval_runtime": 0.6073,
+      "eval_samples_per_second": 489.085,
+      "eval_steps_per_second": 1.647,
+      "step": 168
+    },
+    {
+      "epoch": 85.0,
+      "grad_norm": 0.3931790292263031,
+      "learning_rate": 0.00029474560786952957,
+      "loss": 0.273,
+      "step": 170
+    },
+    {
+      "epoch": 85.0,
+      "eval_loss": 0.3132490813732147,
+      "eval_runtime": 0.6683,
+      "eval_samples_per_second": 444.398,
+      "eval_steps_per_second": 1.496,
+      "step": 170
+    },
+    {
+      "epoch": 86.0,
+      "grad_norm": 0.4931921064853668,
+      "learning_rate": 0.00029462126067537756,
+      "loss": 0.2713,
+      "step": 172
+    },
+    {
+      "epoch": 86.0,
+      "eval_loss": 0.3119468688964844,
+      "eval_runtime": 0.5093,
+      "eval_samples_per_second": 583.176,
+      "eval_steps_per_second": 1.964,
+      "step": 172
+    },
+    {
+      "epoch": 87.0,
+      "grad_norm": 0.5810508131980896,
+      "learning_rate": 0.00029449548612776866,
+      "loss": 0.2665,
+      "step": 174
+    },
+    {
+      "epoch": 87.0,
+      "eval_loss": 0.31254178285598755,
+      "eval_runtime": 0.7282,
+      "eval_samples_per_second": 407.876,
+      "eval_steps_per_second": 1.373,
+      "step": 174
+    },
+    {
+      "epoch": 88.0,
+      "grad_norm": 0.6227275729179382,
+      "learning_rate": 0.00029436828546804686,
+      "loss": 0.2719,
+      "step": 176
+    },
+    {
+      "epoch": 88.0,
+      "eval_loss": 0.3148057162761688,
+      "eval_runtime": 0.7497,
+      "eval_samples_per_second": 396.169,
+      "eval_steps_per_second": 1.334,
+      "step": 176
+    },
+    {
+      "epoch": 89.0,
+      "grad_norm": 0.6895466446876526,
+      "learning_rate": 0.0002942396599516314,
+      "loss": 0.2666,
+      "step": 178
+    },
+    {
+      "epoch": 89.0,
+      "eval_loss": 0.31230518221855164,
+      "eval_runtime": 0.6871,
+      "eval_samples_per_second": 432.239,
+      "eval_steps_per_second": 1.455,
+      "step": 178
+    },
+    {
+      "epoch": 90.0,
+      "grad_norm": 0.6018730401992798,
+      "learning_rate": 0.0002941096108480041,
+      "loss": 0.2641,
+      "step": 180
+    },
+    {
+      "epoch": 90.0,
+      "eval_loss": 0.3139157295227051,
+      "eval_runtime": 0.6814,
+      "eval_samples_per_second": 435.86,
+      "eval_steps_per_second": 1.468,
+      "step": 180
+    },
+    {
+      "epoch": 91.0,
+      "grad_norm": 0.5773816704750061,
+      "learning_rate": 0.0002939781394406971,
+      "loss": 0.2679,
+      "step": 182
+    },
+    {
+      "epoch": 91.0,
+      "eval_loss": 0.314519464969635,
+      "eval_runtime": 0.6242,
+      "eval_samples_per_second": 475.778,
+      "eval_steps_per_second": 1.602,
+      "step": 182
+    },
+    {
+      "epoch": 92.0,
+      "grad_norm": 0.430232971906662,
+      "learning_rate": 0.00029384524702728013,
+      "loss": 0.2636,
+      "step": 184
+    },
+    {
+      "epoch": 92.0,
+      "eval_loss": 0.3126896917819977,
+      "eval_runtime": 0.686,
+      "eval_samples_per_second": 432.948,
+      "eval_steps_per_second": 1.458,
+      "step": 184
+    },
+    {
+      "epoch": 93.0,
+      "grad_norm": 0.5966992974281311,
+      "learning_rate": 0.0002937109349193477,
+      "loss": 0.2662,
+      "step": 186
+    },
+    {
+      "epoch": 93.0,
+      "eval_loss": 0.3127867877483368,
+      "eval_runtime": 0.704,
+      "eval_samples_per_second": 421.861,
+      "eval_steps_per_second": 1.42,
+      "step": 186
+    },
+    {
+      "epoch": 94.0,
+      "grad_norm": 0.5436354875564575,
+      "learning_rate": 0.000293575204442506,
+      "loss": 0.2625,
+      "step": 188
+    },
+    {
+      "epoch": 94.0,
+      "eval_loss": 0.3121253550052643,
+      "eval_runtime": 0.6839,
+      "eval_samples_per_second": 434.267,
+      "eval_steps_per_second": 1.462,
+      "step": 188
+    },
+    {
+      "epoch": 95.0,
+      "grad_norm": 0.6525291800498962,
+      "learning_rate": 0.00029343805693636017,
+      "loss": 0.2643,
+      "step": 190
+    },
+    {
+      "epoch": 95.0,
+      "eval_loss": 0.30872872471809387,
+      "eval_runtime": 0.7643,
+      "eval_samples_per_second": 388.597,
+      "eval_steps_per_second": 1.308,
+      "step": 190
+    },
+    {
+      "epoch": 96.0,
+      "grad_norm": 0.41923660039901733,
+      "learning_rate": 0.0002932994937545007,
+      "loss": 0.2605,
+      "step": 192
+    },
+    {
+      "epoch": 96.0,
+      "eval_loss": 0.30773305892944336,
+      "eval_runtime": 0.621,
+      "eval_samples_per_second": 478.226,
+      "eval_steps_per_second": 1.61,
+      "step": 192
+    },
+    {
+      "epoch": 97.0,
+      "grad_norm": 0.49382027983665466,
+      "learning_rate": 0.0002931595162644901,
+      "loss": 0.2613,
+      "step": 194
+    },
+    {
+      "epoch": 97.0,
+      "eval_loss": 0.3081483542919159,
+      "eval_runtime": 0.5459,
+      "eval_samples_per_second": 544.096,
+      "eval_steps_per_second": 1.832,
+      "step": 194
+    },
+    {
+      "epoch": 98.0,
+      "grad_norm": 0.3870932459831238,
+      "learning_rate": 0.0002930181258478499,
+      "loss": 0.2655,
+      "step": 196
+    },
+    {
+      "epoch": 98.0,
+      "eval_loss": 0.3087702691555023,
+      "eval_runtime": 0.7521,
+      "eval_samples_per_second": 394.875,
+      "eval_steps_per_second": 1.33,
+      "step": 196
+    },
+    {
+      "epoch": 99.0,
+      "grad_norm": 0.3989209532737732,
+      "learning_rate": 0.00029287532390004633,
+      "loss": 0.2605,
+      "step": 198
+    },
+    {
+      "epoch": 99.0,
+      "eval_loss": 0.30880430340766907,
+      "eval_runtime": 0.6471,
+      "eval_samples_per_second": 458.95,
+      "eval_steps_per_second": 1.545,
+      "step": 198
+    },
+    {
+      "epoch": 100.0,
+      "grad_norm": 0.6288752555847168,
+      "learning_rate": 0.00029273111183047697,
+      "loss": 0.2624,
+      "step": 200
+    },
+    {
+      "epoch": 100.0,
+      "eval_loss": 0.30914193391799927,
+      "eval_runtime": 0.7927,
+      "eval_samples_per_second": 374.659,
+      "eval_steps_per_second": 1.261,
+      "step": 200
+    },
+    {
+      "epoch": 101.0,
+      "grad_norm": 0.7080681324005127,
+      "learning_rate": 0.0002925854910624568,
+      "loss": 0.2532,
+      "step": 202
+    },
+    {
+      "epoch": 101.0,
+      "eval_loss": 0.3055001199245453,
+      "eval_runtime": 0.6808,
+      "eval_samples_per_second": 436.251,
+      "eval_steps_per_second": 1.469,
+      "step": 202
+    },
+    {
+      "epoch": 102.0,
+      "grad_norm": 0.6294286251068115,
+      "learning_rate": 0.00029243846303320386,
+      "loss": 0.2603,
+      "step": 204
+    },
+    {
+      "epoch": 102.0,
+      "eval_loss": 0.3056044280529022,
+      "eval_runtime": 0.7558,
+      "eval_samples_per_second": 392.961,
+      "eval_steps_per_second": 1.323,
+      "step": 204
+    },
+    {
+      "epoch": 103.0,
+      "grad_norm": 0.6036813259124756,
+      "learning_rate": 0.0002922900291938255,
+      "loss": 0.2576,
+      "step": 206
+    },
+    {
+      "epoch": 103.0,
+      "eval_loss": 0.30621686577796936,
+      "eval_runtime": 0.7737,
+      "eval_samples_per_second": 383.867,
+      "eval_steps_per_second": 1.292,
+      "step": 206
+    },
+    {
+      "epoch": 104.0,
+      "grad_norm": 0.44163164496421814,
+      "learning_rate": 0.00029214019100930384,
+      "loss": 0.2532,
+      "step": 208
+    },
+    {
+      "epoch": 104.0,
+      "eval_loss": 0.305007666349411,
+      "eval_runtime": 0.7019,
+      "eval_samples_per_second": 423.12,
+      "eval_steps_per_second": 1.425,
+      "step": 208
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 2000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1000,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "EarlyStoppingCallback": {
+      "args": {
+        "early_stopping_patience": 10,
+        "early_stopping_threshold": 1e-05
+      },
+      "attributes": {
+        "early_stopping_patience_counter": 0
+      }
+    },
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1925114350927872.0,
+  "train_batch_size": 512,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:824c3ad32bc9111e70d255f5efe27e031459ce8027f0b36f339a4ff529c01b13
+size 5969