update新结构

2026-01-26 18:27:41 +08:00
parent bc838d7cd7
commit cb610281ce
6 changed files with 55 additions and 2 deletions
--- a/example/config.json
+++ b/example/config.json
@@ -32,6 +32,9 @@
  "model_ff_mult": 2,
  "model_pos_dim": 64,
  "model_use_pos_embed": true,
+  "model_use_feature_graph": true,
+  "feature_graph_scale": 0.1,
+  "feature_graph_dropout": 0.0,
  "disc_mask_scale": 0.9,
  "cont_loss_weighting": "inv_std",
  "cont_loss_eps": 1e-6,