update ks

2026-01-25 18:13:37 +08:00
parent b3c45010a4
commit bc838d7cd7
3 changed files with 47 additions and 6 deletions
--- a/example/config.json
+++ b/example/config.json
@@ -39,6 +39,9 @@
  "cont_clamp_x0": 5.0,
  "quantile_loss_weight": 0.1,
  "quantile_points": [0.05, 0.25, 0.5, 0.75, 0.95],
+  "quantile_loss_warmup_steps": 200,
+  "quantile_loss_clip": 6.0,
+  "quantile_loss_huber_delta": 1.0,
  "shuffle_buffer": 256,
  "sample_batch_size": 8,
  "sample_seq_len": 128