{
  "completed_shards": [
    0,
    1,
    2,
    3,
    4,
    5
  ],
  "last_shard": 5,
  "total_steps": 291,
  "version": 3,
  "shard_losses": {
    "0": 1.2034212966119089,
    "1": 1.0793929558533888,
    "2": 1.021393289932838,
    "3": 0.9929503855796961,
    "4": 0.9694558657132663,
    "5": 0.9570254064523257
  }
}