model:
  params:
    depth: 28
    n_heads: 16
    d_head: 72
    condition_global_keys_list: []
    is_last_feature: 1
    qk_norm: 1
    is_pecode_latent: 1
    is_pecode_latent_pe: 1
    target_mu:
      - -0.24537603557109833
      - -0.7267504930496216
      - -0.0816086083650589
      - -0.17939254641532898
      - 0.1138397604227066
      - -0.8360004425048828
      - 0.47255510091781616
      - -0.07358624786138535
      - 0.1038229838013649
      - -0.07785441726446152
      - 0.7885988354682922
      - -1.165297269821167
      - -0.2916831970214844
      - 0.5662569403648376
      - -0.6377986669540405
      - -0.05221424624323845
      - -0.6274069547653198
      - 0.13630224764347076
      - -0.29677602648735046
      - -0.6062723398208618
      - -0.2247072458267212
      - 0.35355159640312195
      - 0.36897560954093933
      - 0.27751320600509644
      - -0.14971300959587097
      - -0.19294434785842896
      - -0.1479526162147522
      - -0.02374897710978985
      - -0.14040935039520264
      - 0.25790536403656006
      - 0.16871976852416992
      - -0.06627867370843887
    target_std:
      - 2.350555181503296
      - 1.936501145362854
      - 1.0859044790267944
      - 3.1779537200927734
      - 1.4753531217575073
      - 2.520535707473755
      - 2.101593017578125
      - 3.3897223472595215
      - 1.3110147714614868
      - 1.770803451538086
      - 1.8564962148666382
      - 2.5058887004852295
      - 2.455212354660034
      - 2.7981133460998535
      - 2.7846906185150146
      - 2.51442289352417
      - 2.157825469970703
      - 1.4927104711532593
      - 2.9518864154815674
      - 1.0375343561172485
      - 3.0358963012695312
      - 2.821928024291992
      - 1.0281389951705933
      - 2.0190064907073975
      - 2.2814815044403076
      - 2.8889224529266357
      - 0.8938020467758179
      - 2.296116352081299
      - 2.3712480068206787
      - 1.9524204730987549
      - 2.948482036590576
      - 1.6851431131362915