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