Training in progress, step 70, checkpoint

checkpoint-70/config.json

@@ -0,0 +1,24 @@
+{
+  "architectures": [
+    "RNNForLanguageModeling"
+  ],
+  "bidirectional": false,
+  "dropout_p": 0.1,
+  "emb_init_range": 0.1,
+  "embedding_dim": 1024,
+  "embedding_kwargs": {},
+  "hidden_dim": 1024,
+  "lin_init_range": 0.03125,
+  "lm_in_features": 1024,
+  "model_type": "rnn",
+  "num_layers": 2,
+  "output_last_state": false,
+  "output_recurrent_outputs": false,
+  "recur_init_range": 0.03125,
+  "rnn_kwargs": {},
+  "rnn_type": "LSTM",
+  "tie_weights": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.33.3",
+  "vocab_size": 50002
+}

checkpoint-70/optimizer.pt

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

checkpoint-70/pytorch_model.bin

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

checkpoint-70/rng_state.pth

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

checkpoint-70/scheduler.pt

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

checkpoint-70/special_tokens_map.json

@@ -0,0 +1,5 @@
+{
+  "eos_token": "<eos>",
+  "pad_token": "<pad>",
+  "unk_token": "<unk>"
+}

checkpoint-70/tokenizer_config.json

@@ -0,0 +1,12 @@
+{
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<eos>",
+  "max_length": null,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_to_multiple_of": null,
+  "pad_token": "<pad>",
+  "pad_token_type_id": 0,
+  "padding_side": "right",
+  "tokenizer_class": "PreTrainedTokenizerFast",
+  "unk_token": "<unk>"
+}

checkpoint-70/trainer_state.json

@@ -0,0 +1,117 @@
+{
+  "best_metric": 8.271754264831543,
+  "best_model_checkpoint": "/mmfs1/gscratch/stf/abhinavp/corpus-filtering/outputs/fict-full-lstm-42/checkpoints/checkpoint-70",
+  "epoch": 0.56,
+  "eval_steps": 10,
+  "global_step": 70,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.08,
+      "learning_rate": 4.600000000000001e-05,
+      "loss": 10.813,
+      "step": 10
+    },
+    {
+      "epoch": 0.08,
+      "eval_loss": 10.803234100341797,
+      "eval_runtime": 2.8007,
+      "eval_samples_per_second": 357.057,
+      "eval_steps_per_second": 44.632,
+      "step": 10
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 4.2e-05,
+      "loss": 10.793,
+      "step": 20
+    },
+    {
+      "epoch": 0.16,
+      "eval_loss": 10.778646469116211,
+      "eval_runtime": 2.6573,
+      "eval_samples_per_second": 376.32,
+      "eval_steps_per_second": 47.04,
+      "step": 20
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 3.8e-05,
+      "loss": 10.7565,
+      "step": 30
+    },
+    {
+      "epoch": 0.24,
+      "eval_loss": 10.711685180664062,
+      "eval_runtime": 2.8706,
+      "eval_samples_per_second": 348.354,
+      "eval_steps_per_second": 43.544,
+      "step": 30
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 3.4000000000000007e-05,
+      "loss": 10.5835,
+      "step": 40
+    },
+    {
+      "epoch": 0.32,
+      "eval_loss": 10.208820343017578,
+      "eval_runtime": 2.7457,
+      "eval_samples_per_second": 364.201,
+      "eval_steps_per_second": 45.525,
+      "step": 40
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 3e-05,
+      "loss": 9.8124,
+      "step": 50
+    },
+    {
+      "epoch": 0.4,
+      "eval_loss": 9.19725227355957,
+      "eval_runtime": 2.6505,
+      "eval_samples_per_second": 377.282,
+      "eval_steps_per_second": 47.16,
+      "step": 50
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.6000000000000002e-05,
+      "loss": 9.0079,
+      "step": 60
+    },
+    {
+      "epoch": 0.48,
+      "eval_loss": 8.602240562438965,
+      "eval_runtime": 2.6478,
+      "eval_samples_per_second": 377.678,
+      "eval_steps_per_second": 47.21,
+      "step": 60
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.2000000000000003e-05,
+      "loss": 8.5534,
+      "step": 70
+    },
+    {
+      "epoch": 0.56,
+      "eval_loss": 8.271754264831543,
+      "eval_runtime": 2.841,
+      "eval_samples_per_second": 351.994,
+      "eval_steps_per_second": 43.999,
+      "step": 70
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 125,
+  "num_train_epochs": 1,
+  "save_steps": 10,
+  "total_flos": 4923721736640.0,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-70/training_args.bin

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