train_state.json

{
 "stage": 2,
 "elapsed": {
  "1": 32401.568507432938,
  "2": 5401.360342264175
 },
 "step": {
  "1": 3112,
  "2": 806
 },
 "scale": 8192.0,
 "sessions": 1,
 "done_stages": [
  1,
  2
 ],
 "finished": true,
 "backbone": "google/bert_uncased_L-12_H-512_A-8",
 "good": 253
}