{ "d_model": 128, "nhead": 4, "num_layers": 4, "vocab_size": 257, "avg_loss": 0.04798714957871745, "avg_bpb": 0.06923082272361258 }