update

example/results/config_used.json

@@ -36,9 +36,9 @@
   "shuffle_buffer": 256,
   "cont_loss_weighting": "inv_std",
   "cont_loss_eps": 1e-06,
-  "cont_target": "x0",
+  "cont_target": "v",
   "cont_clamp_x0": 5.0,
-  "quantile_loss_weight": 0.3,
+  "quantile_loss_weight": 0.1,
   "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.645508,14001.333008,1.724242
+0,0,11527.787109,16467.474609,1.724242
-0,10,8648.753906,12354.557617,1.050417
+0,10,8925.303711,12749.924805,1.066143
-0,20,6286.167480,8979.179688,0.775981
+0,20,122902.148438,175573.968750,1.048266
-0,30,6297.359375,8995.274414,0.598672
+0,30,27893.091797,39846.773438,1.019252
-0,40,6953.571289,9932.811523,0.402265
+0,40,19384.621094,27692.042969,0.490717
-0,50,5424.526855,7748.514160,0.340582
+0,50,27559.511719,39370.480469,0.440001
-1,0,17800.640625,25428.843750,0.519747
+1,0,983913.437500,1405590.250000,0.672650
-1,10,9939.250977,14198.417969,0.343056
+1,10,15702.782227,22431.642578,1.949623
-1,20,4974.197754,7105.227539,0.211707
+1,20,119934.757812,171335.171875,0.266383
-1,30,5463.194824,7803.769043,0.329701
+1,30,46955.992188,67079.718750,0.451367
-1,40,4324.631348,6177.318359,0.293905
+1,40,14945.393555,21350.371094,0.329458
-1,50,4046.199707,5779.564941,0.276044
+1,50,9970.682617,14243.659180,0.292640
-2,0,15467.764648,22096.150391,0.564124
+2,0,909690.937500,1299558.000000,0.753066
-2,10,7146.419434,10208.712891,0.314471
+2,10,24938.431641,35625.917969,0.833421
-2,20,5792.232422,8273.794922,0.202803
+2,20,133833.140625,191190.000000,0.250991
-2,30,4184.514160,5977.152344,0.308605
+2,30,84536.242188,120765.851562,0.342500
-2,40,3683.652344,5261.627441,0.295088
+2,40,28339.226562,40484.390625,0.374297
-2,50,3847.852295,5496.241211,0.298570
+2,50,12867.059570,18381.312500,0.347191
-3,0,15298.166016,21853.927734,0.546051
+3,0,992322.187500,1417602.625000,0.799355
-3,10,5180.031250,7399.642578,0.307531
+3,10,112537.000000,160766.156250,2.107298
-3,20,4542.523926,6488.400879,0.309543
+3,20,92161.367188,131658.921875,0.221603
-3,30,3677.723877,5253.100098,0.382187
+3,30,58064.828125,82949.531250,0.354935
-3,40,3390.495605,4842.778809,0.282848
+3,40,11155.012695,15935.555664,0.291318
-3,50,3529.083984,5040.805664,0.294322
+3,50,8351.337891,11930.288086,0.319319
-4,0,14632.495117,20902.976562,0.548208
+4,0,985018.437500,1407168.625000,0.989894
-4,10,10205.823242,14579.279297,0.359179
+4,10,150587.375000,215124.531250,0.410451
-4,20,4333.018066,6189.319824,0.188001
+4,20,290905.875000,415579.625000,0.302090
-4,30,3423.444336,4889.942383,0.309461
+4,30,110148.171875,157354.265625,0.438021
-4,40,3474.843018,4963.380859,0.289691
+4,40,22988.332031,32840.261719,0.350128
-4,50,2808.380615,4011.301270,0.295505
+4,50,17114.314453,24448.830078,0.326380
-5,0,12855.440430,18364.394531,0.511693
+5,0,748518.937500,1069312.250000,0.816294
-5,10,5029.373535,7184.408691,0.312978
+5,10,117673.734375,168105.093750,0.387743
-5,20,3587.488037,5124.273438,0.240948
+5,20,320272.375000,457531.718750,0.386736
-5,30,3527.654785,5038.819824,0.320213
+5,30,31867.970703,45525.425781,0.396934
-5,40,3269.149414,4669.565430,0.270011
+5,40,15146.457031,21637.613281,0.302920
-5,50,2967.330566,4238.372559,0.285904
+5,50,10223.821289,14605.273438,0.319003
-6,0,12980.448242,18542.851562,0.522880
+6,0,593388.375000,847697.375000,0.711933
-6,10,4694.814453,6706.399902,0.302162
+6,10,28501.130859,40715.085938,1.744874
-6,20,3500.672119,5000.198730,0.286830
+6,20,211790.203125,302557.187500,0.308398
-6,30,3212.191406,4588.181152,0.304900
+6,30,16399.605469,23427.794922,0.357711
-6,40,3307.898193,4724.925293,0.267466
+6,40,11942.860352,17061.058594,0.288670
-6,50,3085.380859,4407.030273,0.265367
+6,50,8927.477539,12753.364258,0.288999
-7,0,4270.843262,6100.737793,0.471359
+7,0,535968.187500,765668.375000,0.773959
-7,10,4797.216309,6852.770020,0.320178
+7,10,12762.362305,18231.421875,1.077526
-7,20,3258.792969,4654.722168,0.244809
+7,20,42435.332031,60621.671875,0.316750
-7,30,3169.457031,4527.143555,0.293432
+7,30,15629.781250,22328.056641,0.334986
-7,40,2939.750488,4198.982910,0.287075
+7,40,10731.666992,15330.766602,0.313070
-7,50,2996.508301,4280.066895,0.251909
+7,50,8851.616211,12644.987305,0.279095
-8,0,5548.325195,7925.604492,0.489300
+8,0,365685.375000,522407.281250,0.723811
-8,10,4157.997559,5939.641113,0.304927
+8,10,10646.472656,15208.710938,1.137090
-8,20,3767.288330,5381.098145,0.210738
+8,20,30592.496094,43703.382812,0.223082
-8,30,2840.536377,4057.270020,0.276704
+8,30,26734.046875,38191.308594,0.311048
-8,40,3035.344727,4335.562500,0.257632
+8,40,11779.257812,16827.343750,0.270093
-8,50,3004.891357,4292.053223,0.266179
+8,50,9673.774414,13819.486328,0.311987
-9,0,3490.820312,4986.386719,0.471640
+9,0,126674.117188,180962.640625,0.731923
-9,10,4166.912598,5952.338379,0.343747
+9,10,8981.769531,12830.742188,0.734161
-9,20,3064.893555,4377.696777,0.228080
+9,20,23218.923828,33169.687500,0.278144
-9,30,2888.233154,4125.366699,0.271929
+9,30,19692.558594,28132.021484,0.300332
-9,40,2755.770752,3936.188721,0.260033
+9,40,12912.096680,18445.681641,0.284465
-9,50,2579.736084,3684.677002,0.268067
+9,50,9004.684570,12863.654297,0.308479