update

example/results/config_used.json

@@ -38,6 +38,12 @@
   "cont_loss_eps": 1e-06,
   "cont_target": "x0",
   "cont_clamp_x0": 5.0,
+  "use_temporal_stage1": true,
+  "temporal_hidden_dim": 256,
+  "temporal_num_layers": 1,
+  "temporal_dropout": 0.0,
+  "temporal_epochs": 2,
+  "temporal_lr": 0.001,
   "sample_batch_size": 8,
   "sample_seq_len": 128
 }

example/results/metrics_history.csv

@@ -0,0 +1,3 @@
+timestamp,avg_ks,avg_jsd,avg_lag1_diff
+2026-01-26T14:24:11.947556,0.7131992629716979,0.0327603185748291,0.23276327008767583
+2026-01-26T14:24:11.996556,0.7131992629716979,0.0327603185748291,0.23276327008767583

example/results/train_log.csv

@@ -1,61 +1,61 @@
 epoch,step,loss,loss_cont,loss_disc
-0,0,9801.450195,14001.333008,1.724242
-0,10,8648.388672,12354.390625,1.050415
-0,20,6285.666992,8979.191406,0.775979
-0,30,6296.939941,8995.372070,0.598658
-0,40,7126.128906,10180.011719,0.402291
-0,50,5381.071289,7687.099121,0.340463
-1,0,17876.904297,25538.212891,0.522857
-1,10,9174.909180,13106.868164,0.338315
-1,20,7635.462891,10907.713867,0.211883
-1,30,5425.212402,7750.165039,0.323606
-1,40,4372.716309,6246.610840,0.296102
-1,50,3846.437988,5494.793945,0.273941
-2,0,17057.958984,24368.269531,0.564671
-2,10,7089.009766,10127.021484,0.315390
-2,20,4230.856445,6043.994141,0.201429
-2,30,3744.107910,5348.593262,0.309118
-2,40,3531.041992,5044.219238,0.295744
-2,50,3570.459229,5100.528320,0.297530
-3,0,14367.601562,20524.917969,0.529623
-3,10,6734.334473,9620.347656,0.304721
-3,20,6140.179688,8771.602539,0.193836
-3,30,4089.454102,5841.940918,0.317864
-3,40,3553.830811,5076.785645,0.269531
-3,50,3590.448242,5129.088867,0.287063
-4,0,14410.816406,20586.648438,0.543822
-4,10,6411.443359,9159.058594,0.341742
-4,20,3816.795166,5452.479492,0.198213
-4,30,4069.170898,5812.959961,0.329676
-4,40,3484.921631,4978.332520,0.296284
-4,50,2802.801514,4003.873779,0.299286
-5,0,13335.293945,19050.201172,0.509557
-5,10,5531.156738,7901.527344,0.293409
-5,20,3844.260010,5491.696777,0.241263
-5,30,3619.303223,5170.297363,0.317237
-5,40,3492.172852,4988.697754,0.281641
-5,50,3069.457275,4384.815918,0.287269
-6,0,8740.982422,12486.912109,0.483061
-6,10,6110.571777,8729.239258,0.347929
-6,20,3350.194092,4785.889160,0.239650
-6,30,3008.237549,4297.353516,0.300327
-6,40,2944.483887,4206.288574,0.273457
-6,50,3018.033447,4311.365234,0.259749
-7,0,6784.341309,9691.704102,0.495378
-7,10,4946.872559,7066.822754,0.321541
-7,20,2816.704102,4023.745361,0.274515
-7,30,2991.350830,4273.229980,0.299238
-7,40,3023.450684,4319.083984,0.307114
-7,50,2944.912598,4206.909668,0.253002
-8,0,7364.851562,10521.002930,0.498283
-8,10,3897.301025,5567.439453,0.311517
-8,20,3313.474854,4733.448242,0.203364
-8,30,2697.139648,3852.940430,0.271355
-8,40,2955.225342,4221.633301,0.273175
-8,50,2932.081787,4188.575684,0.262651
-9,0,4065.651611,5807.854004,0.513017
-9,10,4358.108398,6225.728516,0.329111
-9,20,3417.019043,4881.362305,0.218488
-9,30,2917.136719,4167.226562,0.260805
-9,40,2786.277832,3980.287109,0.256345
-9,50,2602.660645,3717.971680,0.268723
+0,0,10076.751953,14394.618164,1.730766
+0,10,8327.669922,11896.221680,1.047697
+0,20,6480.192383,9257.069336,0.813652
+0,30,6247.419434,8924.612305,0.635893
+0,40,5726.993652,8181.236328,0.427442
+0,50,5137.051270,7338.496582,0.345255
+1,0,17764.189453,25377.191406,0.519132
+1,10,7646.325195,10923.160156,0.377619
+1,20,5722.854004,8175.404785,0.235613
+1,30,4850.348145,6928.932129,0.318460
+1,40,3980.531982,5686.357910,0.271540
+1,50,3617.966797,5168.394043,0.303173
+2,0,14719.765625,21028.019531,0.507699
+2,10,4538.907715,6484.002441,0.352795
+2,20,4042.695801,5775.192383,0.203912
+2,30,3406.538330,4866.341309,0.331246
+2,40,3423.300781,4890.292969,0.319031
+2,50,3576.180176,5108.698242,0.305018
+3,0,9969.816406,14242.361328,0.546363
+3,10,4930.293945,7043.129883,0.343653
+3,20,4363.328125,6233.225098,0.236356
+3,30,3856.120605,5508.612793,0.305913
+3,40,3440.814453,4915.323730,0.293026
+3,50,3633.039551,5189.932617,0.289079
+4,0,11934.705078,17049.355469,0.521082
+4,10,4701.100586,6715.739258,0.277104
+4,20,3558.578125,5083.573242,0.255970
+4,30,3005.302734,4293.161133,0.299914
+4,40,3156.410645,4509.035645,0.285384
+4,50,2880.248047,4114.524902,0.268697
+5,0,5173.457520,7390.440918,0.496033
+5,10,3827.181152,5467.258789,0.333715
+5,20,3353.919922,4791.204590,0.255503
+5,30,3278.056396,4682.817383,0.280490
+5,40,3599.985596,5142.721680,0.268798
+5,50,2784.245605,3977.385742,0.252457
+6,0,5414.885254,7735.326660,0.523181
+6,10,4215.503418,6022.016602,0.306116
+6,20,3488.155273,4982.981445,0.227785
+6,30,2953.763916,4219.535645,0.297097
+6,40,2894.835693,4135.359375,0.280651
+6,50,2884.175293,4120.137207,0.264586
+7,0,6498.497070,9283.343750,0.522994
+7,10,4186.162598,5980.098145,0.312845
+7,20,2971.318848,4244.639160,0.238725
+7,30,2816.141113,4022.941406,0.273892
+7,40,2713.621826,3876.488281,0.267141
+7,50,2975.131592,4250.065430,0.286403
+8,0,6809.148438,9727.107422,0.577808
+8,10,3750.679443,5357.983398,0.303163
+8,20,3877.942383,5539.823242,0.220803
+8,30,2785.791992,3979.580322,0.285388
+8,40,2866.808105,4095.317871,0.285348
+8,50,2852.916016,4075.487793,0.249002
+9,0,4856.940430,6938.258301,0.531997
+9,10,3285.523438,4693.470215,0.313796
+9,20,3039.602539,4342.197754,0.213819
+9,30,2918.992432,4169.859375,0.303131
+9,40,2778.509521,3969.181641,0.275007
+9,50,2612.647949,3732.237061,0.273503