{
  "completed_shards": [
    0
  ],
  "last_shard": 0,
  "total_steps": 16,
  "version": 3,
  "shard_losses": {
    "0": 1.3516322299838066
  }
}