1 files changed, 70 insertions, 0 deletions
diff --git a/research/flossing/flossing_suite/results/smoke/trm_task_batch32_smoke.json b/research/flossing/flossing_suite/results/smoke/trm_task_batch32_smoke.json
new file mode 100644
index 0000000..c6df53e
--- /dev/null
+++ b/research/flossing/flossing_suite/results/smoke/trm_task_batch32_smoke.json
@@ -0,0 +1,70 @@
+{
+  "args": {
+    "model": "trm",
+    "ckpt_root": "/home/yurenh2/rrm/trm/checkpoints/Sudoku-extreme-1k-aug-1000-ACT-torch/pretrain_mlp_t_sudoku_official_gbs768_repro",
+    "ckpt_name": "__random__",
+    "train_steps": 1,
+    "batch_size": 32,
+    "task_batch_size": 32,
+    "floss_batch_size": 4,
+    "train_lr": 0.0001,
+    "floss_lr": 0.0001,
+    "floss_steps": 0,
+    "interfloss_at": "",
+    "interfloss_every": 0,
+    "interfloss_start": 0,
+    "interfloss_stop": -1,
+    "floss_mode": "engelken_l2",
+    "lambda_star": 0.0,
+    "k_lyap": 4,
+    "lyap_act_steps": 4,
+    "lyap_start_act": 0,
+    "seed": 43,
+    "eval_every": 1,
+    "eval_n": 8,
+    "eval_batch_size": 8,
+    "floss_log_every": 1,
+    "eval_after_floss": false,
+    "kl_beta": 0.0,
+    "kl_replay_size": 64,
+    "kl_batch_size": 8,
+    "kl_temperature": 1.0,
+    "init_seed": 123,
+    "train_puzzle_emb": true,
+    "puzzle_emb_lr": 0.0001,
+    "puzzle_emb_weight_decay": 1.0,
+    "out": "research/flossing/flossing_suite/results/smoke/trm_task_batch32_smoke.json"
+  },
+  "initial_acc": 0.0,
+  "initial_tok_acc": 0.07561728395061729,
+  "interfloss_steps": [],
+  "task_steps": [
+    {
+      "train_step": 1,
+      "sup_loss": 2.581977193130616
+    }
+  ],
+  "floss_episodes": [],
+  "evals": [
+    {
+      "kind": "initial",
+      "train_step": 0,
+      "acc": 0.0,
+      "tok_acc": 0.07561728395061729
+    },
+    {
+      "kind": "task",
+      "train_step": 1,
+      "acc": 0.0,
+      "tok_acc": 0.09722222222222222
+    },
+    {
+      "kind": "final",
+      "train_step": 1,
+      "acc": 0.0,
+      "tok_acc": 0.09722222222222222
+    }
+  ],
+  "final_acc": 0.0,
+  "final_tok_acc": 0.09722222222222222
+}
+\ No newline at end of file