update

example/results/config_used.json

@@ -38,7 +38,7 @@
   "cont_loss_eps": 1e-06,
   "cont_target": "x0",
   "cont_clamp_x0": 5.0,
-  "quantile_loss_weight": 0.1,
+  "quantile_loss_weight": 0.3,
   "quantile_points": [
     0.05,
     0.25,

example/results/train_log.csv

@@ -1,61 +1,61 @@
 epoch,step,loss,loss_cont,loss_disc
-0,0,9801.515625,14001.333008,1.724242
+0,0,9801.645508,14001.333008,1.724242
-0,10,8648.471680,12354.386719,1.050416
+0,10,8648.753906,12354.557617,1.050417
-0,20,6285.875000,8979.247070,0.775979
+0,20,6286.167480,8979.179688,0.775981
-0,30,6297.153809,8995.444336,0.598639
+0,30,6297.359375,8995.274414,0.598672
-0,40,7421.842285,10602.228516,0.402729
+0,40,6953.571289,9932.811523,0.402265
-0,50,8768.145508,12525.551758,0.339361
+0,50,5424.526855,7748.514160,0.340582
-1,0,17362.210938,24802.792969,0.521414
+1,0,17800.640625,25428.843750,0.519747
-1,10,9998.125000,14282.763672,0.346019
+1,10,9939.250977,14198.417969,0.343056
-1,20,6025.115234,8606.978516,0.211357
+1,20,4974.197754,7105.227539,0.211707
-1,30,5529.321777,7898.669922,0.321453
+1,30,5463.194824,7803.769043,0.329701
-1,40,4561.373047,6515.895508,0.306940
+1,40,4324.631348,6177.318359,0.293905
-1,50,4133.944336,5905.306641,0.271862
+1,50,4046.199707,5779.564941,0.276044
-2,0,18390.314453,26271.496094,0.559547
+2,0,15467.764648,22096.150391,0.564124
-2,10,5629.449707,8041.824219,0.315514
+2,10,7146.419434,10208.712891,0.314471
-2,20,5088.290527,7268.675293,0.209568
+2,20,5792.232422,8273.794922,0.202803
-2,30,4109.000000,5869.645020,0.316234
+2,30,4184.514160,5977.152344,0.308605
-2,40,4135.112305,5906.964355,0.288305
+2,40,3683.652344,5261.627441,0.295088
-2,50,3574.877686,5106.638672,0.300608
+2,50,3847.852295,5496.241211,0.298570
-3,0,16117.513672,23024.658203,0.541584
+3,0,15298.166016,21853.927734,0.546051
-3,10,7299.515137,10427.646484,0.307387
+3,10,5180.031250,7399.642578,0.307531
-3,20,3928.072998,5611.221680,0.202967
+3,20,4542.523926,6488.400879,0.309543
-3,30,4386.781250,6266.497559,0.325895
+3,30,3677.723877,5253.100098,0.382187
-3,40,3627.836182,5182.310059,0.271281
+3,40,3390.495605,4842.778809,0.282848
-3,50,3513.194580,5018.536133,0.291253
+3,50,3529.083984,5040.805664,0.294322
-4,0,14937.586914,21339.070312,0.552027
+4,0,14632.495117,20902.976562,0.548208
-4,10,6087.895508,8696.762695,0.334634
+4,10,10205.823242,14579.279297,0.359179
-4,20,3961.117676,5658.443359,0.210930
+4,20,4333.018066,6189.319824,0.188001
-4,30,3405.418457,4864.548828,0.317895
+4,30,3423.444336,4889.942383,0.309461
-4,40,3483.671631,4976.360840,0.296246
+4,40,3474.843018,4963.380859,0.289691
-4,50,2833.118164,4047.002441,0.297401
+4,50,2808.380615,4011.301270,0.295505
-5,0,12412.599609,17731.978516,0.492310
+5,0,12855.440430,18364.394531,0.511693
-5,10,4952.285156,7074.484375,0.294987
+5,10,5029.373535,7184.408691,0.312978
-5,20,4023.841309,5748.041504,0.219470
+5,20,3587.488037,5124.273438,0.240948
-5,30,3416.583740,4880.517090,0.310916
+5,30,3527.654785,5038.819824,0.320213
-5,40,3283.848389,4690.912598,0.274809
+5,40,3269.149414,4669.565430,0.270011
-5,50,2953.851807,4219.480469,0.286249
+5,50,2967.330566,4238.372559,0.285904
-6,0,9772.470703,13960.334961,0.525315
+6,0,12980.448242,18542.851562,0.522880
-6,10,4856.467773,6937.604980,0.294435
+6,10,4694.814453,6706.399902,0.302162
-6,20,3487.783203,4982.249023,0.249389
+6,20,3500.672119,5000.198730,0.286830
-6,30,2907.010498,4152.563965,0.302258
+6,30,3212.191406,4588.181152,0.304900
-6,40,2978.796875,4255.132324,0.272516
+6,40,3307.898193,4724.925293,0.267466
-6,50,2954.490723,4220.402832,0.260484
+6,50,3085.380859,4407.030273,0.265367
-7,0,5634.914062,8049.578125,0.477380
+7,0,4270.843262,6100.737793,0.471359
-7,10,4834.394531,6906.059570,0.314162
+7,10,4797.216309,6852.770020,0.320178
-7,20,2799.942871,3999.613770,0.267521
+7,20,3258.792969,4654.722168,0.244809
-7,30,2899.989990,4142.533203,0.294986
+7,30,3169.457031,4527.143555,0.293432
-7,40,2961.559570,4230.455078,0.337903
+7,40,2939.750488,4198.982910,0.287075
-7,50,3053.434814,4361.746094,0.263959
+7,50,2996.508301,4280.066895,0.251909
-8,0,5015.993652,7165.385742,0.495864
+8,0,5548.325195,7925.604492,0.489300
-8,10,3965.379639,5664.615234,0.316581
+8,10,4157.997559,5939.641113,0.304927
-8,20,3669.429688,5241.729980,0.204997
+8,20,3767.288330,5381.098145,0.210738
-8,30,2815.938232,4022.476562,0.271451
+8,30,2840.536377,4057.270020,0.276704
-8,40,2967.452881,4238.926758,0.263370
+8,40,3035.344727,4335.562500,0.257632
-8,50,2930.122314,4185.593262,0.262396
+8,50,3004.891357,4292.053223,0.266179
-9,0,4364.022461,6233.995605,0.496020
+9,0,3490.820312,4986.386719,0.471640
-9,10,4222.906250,6032.508301,0.319415
+9,10,4166.912598,5952.338379,0.343747
-9,20,3070.776367,4386.530762,0.233229
+9,20,3064.893555,4377.696777,0.228080
-9,30,2839.424805,4056.029785,0.265954
+9,30,2888.233154,4125.366699,0.271929
-9,40,2770.363770,3957.375000,0.264960
+9,40,2755.770752,3936.188721,0.260033
-9,50,2557.437256,3653.188721,0.265549
+9,50,2579.736084,3684.677002,0.268067