args.json

{
  "codebook_size": 1024,
  "pose_dims": [
    178,
    3
  ],
  "num_codebooks": 8,
  "hidden_dim": 512,
  "nhead": 16,
  "dim_feedforward": 2048,
  "num_layers": 8
}