embedding_dim: 256
num_kp: 10
num_w: 5