step_540000/config.json

{
  "d_model": 128,
  "nhead": 4,
  "num_layers": 4,
  "vocab_size": 257,
  "avg_loss": 0.038902236989692404,
  "avg_bpb": 0.05612406438451649
}