Training in progress, step 7000

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ade63866bbc493dfbfb03d439072fb9d488f77db218d611ed1e217f724a99068
+oid sha256:f50fcbeac21ddcf83790ff6bbb28d8f29f47fb2d1e4d3dc8a614bfd15b631428
 size 2329638768

tmp-checkpoint-2100/config.json

@@ -0,0 +1,33 @@
+{
+  "_name_or_path": "google/mt5-base",
+  "architectures": [
+    "MT5ForConditionalGeneration"
+  ],
+  "classifier_dropout": 0.0,
+  "d_ff": 2048,
+  "d_kv": 64,
+  "d_model": 768,
+  "decoder_start_token_id": 0,
+  "dense_act_fn": "gelu_new",
+  "dropout_rate": 0.1,
+  "eos_token_id": 1,
+  "feed_forward_proj": "gated-gelu",
+  "initializer_factor": 1.0,
+  "is_encoder_decoder": true,
+  "is_gated_act": true,
+  "layer_norm_epsilon": 1e-06,
+  "model_type": "mt5",
+  "num_decoder_layers": 12,
+  "num_heads": 12,
+  "num_layers": 12,
+  "output_past": true,
+  "pad_token_id": 0,
+  "relative_attention_max_distance": 128,
+  "relative_attention_num_buckets": 32,
+  "tie_word_embeddings": false,
+  "tokenizer_class": "T5Tokenizer",
+  "torch_dtype": "float32",
+  "transformers_version": "4.39.0.dev0",
+  "use_cache": true,
+  "vocab_size": 250112
+}

tmp-checkpoint-2100/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "decoder_start_token_id": 0,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.39.0.dev0"
+}

tmp-checkpoint-2100/model.safetensors

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

tmp-checkpoint-2100/optimizer.pt

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

tmp-checkpoint-2100/rng_state.pth

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

tmp-checkpoint-2100/scheduler.pt

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

tmp-checkpoint-2100/special_tokens_map.json

@@ -0,0 +1,23 @@
+{
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tmp-checkpoint-2100/spiece.model

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

tmp-checkpoint-2100/tokenizer.json

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

tmp-checkpoint-2100/tokenizer_config.json

@@ -0,0 +1,39 @@
+{
+  "add_prefix_space": true,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [],
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "</s>",
+  "extra_ids": 0,
+  "legacy": true,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "T5Tokenizer",
+  "unk_token": "<unk>"
+}

tmp-checkpoint-2100/trainer_state.json

@@ -0,0 +1,73 @@
+{
+  "best_metric": 1.6017142534255981,
+  "best_model_checkpoint": "arbitropy/mt5-base-bcoqa/checkpoint-2100",
+  "epoch": 0.060544904137235116,
+  "eval_steps": 700,
+  "global_step": 2100,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.01,
+      "grad_norm": 14.746796607971191,
+      "learning_rate": 4.9639613665849795e-05,
+      "loss": 4.3627,
+      "step": 500
+    },
+    {
+      "epoch": 0.02,
+      "eval_loss": 1.7135968208312988,
+      "eval_runtime": 62.9933,
+      "eval_samples_per_second": 75.183,
+      "eval_steps_per_second": 18.796,
+      "step": 700
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 3.6309075355529785,
+      "learning_rate": 4.927922733169959e-05,
+      "loss": 2.2652,
+      "step": 1000
+    },
+    {
+      "epoch": 0.04,
+      "eval_loss": 1.6376169919967651,
+      "eval_runtime": 63.2403,
+      "eval_samples_per_second": 74.889,
+      "eval_steps_per_second": 18.722,
+      "step": 1400
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 6.434725761413574,
+      "learning_rate": 4.891884099754937e-05,
+      "loss": 2.0917,
+      "step": 1500
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 10.990394592285156,
+      "learning_rate": 4.8558454663399166e-05,
+      "loss": 1.9203,
+      "step": 2000
+    },
+    {
+      "epoch": 0.06,
+      "eval_loss": 1.6017142534255981,
+      "eval_runtime": 63.1369,
+      "eval_samples_per_second": 75.012,
+      "eval_steps_per_second": 18.753,
+      "step": 2100
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 69370,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 700,
+  "total_flos": 1.956696216451891e+16,
+  "train_batch_size": 4,
+  "trial_name": null,
+  "trial_params": null
+}

tmp-checkpoint-2100/training_args.bin

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5bd5b87bc40502d3f7df1436b012dd996beac219bdea584487f441f0232b5e11
+oid sha256:87723fa72dd0500c3ac83696346c3aa13e6f3d19b13e24891f78d7aff97351f8
 size 5048