default_model: m1

m1:
  arch: moe_1
  embed_dim: 128
  block: 4
  head: 4
  num_mha: 2
  num_ffn: 2
  steps: 2
  model_save_path: "models/TF/M1.h5"