1. Change the default `--dump_tensor_debug_mode` flag value to
`FULL_HEALTH`, a mode more suitable for numerical instability
debugging for the particular bug in this example than the previous
default value `NO_TENSOR`.
2. Change the default `--dump_circular_buffer_size` value to -1, to
accommodate the possibility of longer runs where user would
want to see the debug data in its entirety.
3. Rename a few weight variables. They were previously named in a
confusing way.
4. Change "logits" to "probs", as they are generated by a `softmax`
operation and hence are more accurately described as probability
scores.
PiperOrigin-RevId: 316473164
Change-Id: I4eb13f9581a4d4e550b3b3a5cd132eeffc7dd043
06b1b45e42