From 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Sat, 13 Jun 2026 12:35:36 -0500 Subject: rrm workspace: TRM/HRM/SRM code, Maze dataset, dynamical-analysis pipeline Curated export for clone-and-run Maze training (2x A6000) + diagnostics. trm/hrm pretrain.py carry trajectory-augmentation code (backward-compatible). Heavy artifacts (checkpoints/wandb/npz) gitignored; see PROVENANCE.md. Co-Authored-By: Claude Fable 5 --- research/flossing/engelken_python/gpu_smoke.json | 71 ++++++++++++++++++++++++ 1 file changed, 71 insertions(+) create mode 100644 research/flossing/engelken_python/gpu_smoke.json (limited to 'research/flossing/engelken_python/gpu_smoke.json') diff --git a/research/flossing/engelken_python/gpu_smoke.json b/research/flossing/engelken_python/gpu_smoke.json new file mode 100644 index 0000000..2b4b298 --- /dev/null +++ b/research/flossing/engelken_python/gpu_smoke.json @@ -0,0 +1,71 @@ +{ + "config": { + "hidden_size": 16, + "train_epochs": 2, + "inter_period": 100, + "inter_epochs": 0, + "pre_epochs": 1, + "max_inter_episodes": 0, + "batch_size": 4, + "input_dim": 1, + "train_steps": 40, + "lyap_steps": 24, + "floss_input_steps": 40, + "seed_ic": 1, + "seed_input": 1, + "seed_net": 1, + "seed_ons": 1, + "lr": 0.001, + "beta1": 0.9, + "beta2": 0.999, + "init_type": 1, + "recurrent_gain": 1.0, + "recurrent_mean_gain": 0.0, + "input_scale": 1.0, + "delay": 4, + "ws_std": 1.0, + "ws_mean": 0.0, + "wr_std": 1.0, + "wr_mean": 0.0, + "b_std": 0.1, + "b_mean": 0.0, + "n_lyap": 4, + "task": -1, + "lyap_target": 0.0, + "eval_every": 1, + "eval_batches": 1, + "log_every_floss": 1, + "device": "cuda", + "out": "research/flossing/engelken_python/gpu_smoke.json" + }, + "evals": [ + { + "epoch": 1, + "train_loss": 0.936610221862793, + "train_accuracy": 0.5378787878787878, + "eval_loss": 0.8679313659667969, + "eval_accuracy": 0.5151515151515151, + "elapsed": 1.3251755237579346 + }, + { + "epoch": 2, + "train_loss": 1.0361249446868896, + "train_accuracy": 0.4318181818181818, + "eval_loss": 0.9058582782745361, + "eval_accuracy": 0.49242424242424243, + "elapsed": 1.3534562587738037 + } + ], + "floss": [ + { + "epoch": 1, + "kind": "pre", + "floss_step": 1, + "loss": 0.12374763190746307, + "lambda_mean": -0.31238266825675964, + "lambda_1": -0.08050862699747086, + "elapsed": 1.2431800365447998 + } + ], + "task": [] +} \ No newline at end of file -- cgit v1.2.3