checkpoint step 9000

training-runs/latest/args.json

@@ -8,17 +8,17 @@
   "block_size": 512,
   "batch_size": 4,
   "grad_accum": 8,
-  "max_steps": 8000,
+  "max_steps": 20000,
   "lr": 0.0003,
   "min_lr": 3e-05,
-  "warmup_steps": 500,
+  "warmup_steps": 100,
   "weight_decay": 0.1,
   "grad_clip": 1.0,
-  "eval_interval": 500,
+  "dropout": 0.1,
+  "eval_interval": 1000,
   "eval_batches": 20,
-  "save_interval": 500,
+  "save_interval": 1000,
   "log_interval": 50,
-  "dropout": 0.1,
   "pcs_a": 0.8309193524478643,
   "pcs_b": 0.0,
   "out_dir": "runs/latest",

training-runs/latest/checkpoint_best.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d22c0cfeeceeda651217889358a660945039319a13cd3878c2c284630a561b5a
-size 614210767
+oid sha256:af886a705d1e5936e05867c02d3ae36e2a5ad001c96278a745db3be70cea1a80
+size 614210639

training-runs/latest/checkpoint_last.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4dfe5a8c0ed41b1fdf86edbf49196606f596d7c71240740ee144b9c6586f4e70
-size 614210767
+oid sha256:0424e4fa18e9c58ae28ab27f08905c1bf89999e319ced1647641e2457a77beec
+size 614210639