[
  {
    "step": 4,
    "epoch": 1,
    "batch_in_epoch": 3,
    "theta1": 0.003416,
    "theta2": 0.006961,
    "loss": 2.294372,
    "eta2d_x": 0.478242,
    "eta2d_y": 0.878228,
    "speed": 2e-05,
    "heading_rad": 0.876388,
    "drift_mean": 0.0,
    "drift_var": 1.0,
    "opacity_var": 1.0,
    "photosphere_crossed": false
  },
  {
    "step": 8,
    "epoch": 1,
    "batch_in_epoch": 7,
    "theta1": 0.003463,
    "theta2": 0.007027,
    "loss": 2.284523,
    "eta2d_x": 0.541022,
    "eta2d_y": 0.841008,
    "speed": 2.4e-05,
    "heading_rad": 1.001312,
    "drift_mean": -0.00246215,
    "drift_var": 8.6598e-06,
    "opacity_var": 1.0,
    "photosphere_crossed": false
  },
  {
    "step": 12,
    "epoch": 1,
    "batch_in_epoch": 11,
    "theta1": 0.003531,
    "theta2": 0.007103,
    "loss": 2.267452,
    "eta2d_x": 0.597661,
    "eta2d_y": 0.801749,
    "speed": 3.2e-05,
    "heading_rad": 1.067485,
    "drift_mean": -0.00426763,
    "drift_var": 1.77149e-05,
    "opacity_var": 1.27643e-05,
    "photosphere_crossed": true
  },
  {
    "step": 16,
    "epoch": 1,
    "batch_in_epoch": 15,
    "theta1": 0.003607,
    "theta2": 0.00722,
    "loss": 2.250622,
    "eta2d_x": 0.577899,
    "eta2d_y": 0.816108,
    "speed": 4.1e-05,
    "heading_rad": 0.856328,
    "drift_mean": -0.00420755,
    "drift_var": 1.55906e-05,
    "opacity_var": 1.48363e-05,
    "photosphere_crossed": true
  },
  {
    "step": 20,
    "epoch": 1,
    "batch_in_epoch": 19,
    "theta1": 0.00372,
    "theta2": 0.007352,
    "loss": 2.228747,
    "eta2d_x": 0.602107,
    "eta2d_y": 0.798415,
    "speed": 4.8e-05,
    "heading_rad": 1.008063,
    "drift_mean": -0.00546873,
    "drift_var": 4.92815e-05,
    "opacity_var": 3.04857e-05,
    "photosphere_crossed": true
  },
  {
    "step": 24,
    "epoch": 1,
    "batch_in_epoch": 23,
    "theta1": 0.003803,
    "theta2": 0.007529,
    "loss": 2.207914,
    "eta2d_x": 0.557122,
    "eta2d_y": 0.830431,
    "speed": 5.1e-05,
    "heading_rad": 1.126286,
    "drift_mean": -0.00520825,
    "drift_var": 0.0002509403,
    "opacity_var": 0.0001348761,
    "photosphere_crossed": true
  },
  {
    "step": 28,
    "epoch": 1,
    "batch_in_epoch": 27,
    "theta1": 0.003886,
    "theta2": 0.007723,
    "loss": 2.1637,
    "eta2d_x": 0.521847,
    "eta2d_y": 0.853039,
    "speed": 5.3e-05,
    "heading_rad": 1.210445,
    "drift_mean": -0.0110535,
    "drift_var": 0.0004234127,
    "opacity_var": 0.0003138737,
    "photosphere_crossed": false
  },
  {
    "step": 32,
    "epoch": 1,
    "batch_in_epoch": 31,
    "theta1": 0.003983,
    "theta2": 0.007921,
    "loss": 2.099986,
    "eta2d_x": 0.506701,
    "eta2d_y": 0.862122,
    "speed": 6.2e-05,
    "heading_rad": 1.002411,
    "drift_mean": -0.01592863,
    "drift_var": 0.0004159059,
    "opacity_var": 0.0003962729,
    "photosphere_crossed": false
  },
  {
    "step": 36,
    "epoch": 1,
    "batch_in_epoch": 35,
    "theta1": 0.004099,
    "theta2": 0.008107,
    "loss": 2.070654,
    "eta2d_x": 0.510638,
    "eta2d_y": 0.859796,
    "speed": 4.8e-05,
    "heading_rad": 1.168123,
    "drift_mean": -0.00733304,
    "drift_var": 0.0007139912,
    "opacity_var": 0.0005231897,
    "photosphere_crossed": false
  },
  {
    "step": 40,
    "epoch": 1,
    "batch_in_epoch": 39,
    "theta1": 0.004186,
    "theta2": 0.008267,
    "loss": 1.955794,
    "eta2d_x": 0.506886,
    "eta2d_y": 0.862013,
    "speed": 4.7e-05,
    "heading_rad": 0.888454,
    "drift_mean": -0.02871481,
    "drift_var": 0.0016501436,
    "opacity_var": 0.0011560779,
    "photosphere_crossed": false
  },
  {
    "step": 44,
    "epoch": 1,
    "batch_in_epoch": 43,
    "theta1": 0.004314,
    "theta2": 0.008343,
    "loss": 1.78902,
    "eta2d_x": 0.542831,
    "eta2d_y": 0.839842,
    "speed": 3.9e-05,
    "heading_rad": 0.268789,
    "drift_mean": -0.04169369,
    "drift_var": 0.0011335491,
    "opacity_var": 0.0015700911,
    "photosphere_crossed": false
  },
  {
    "step": 48,
    "epoch": 1,
    "batch_in_epoch": 47,
    "theta1": 0.004509,
    "theta2": 0.008405,
    "loss": 1.642408,
    "eta2d_x": 0.600394,
    "eta2d_y": 0.799704,
    "speed": 5e-05,
    "heading_rad": 0.271644,
    "drift_mean": -0.03665301,
    "drift_var": 0.0001034047,
    "opacity_var": 0.0010773803,
    "photosphere_crossed": false
  },
  {
    "step": 52,
    "epoch": 1,
    "batch_in_epoch": 51,
    "theta1": 0.00479,
    "theta2": 0.008483,
    "loss": 1.366253,
    "eta2d_x": 0.666088,
    "eta2d_y": 0.745873,
    "speed": 9.3e-05,
    "heading_rad": 0.257736,
    "drift_mean": -0.06903866,
    "drift_var": 0.0015003361,
    "opacity_var": 0.0010490029,
    "photosphere_crossed": false
  },
  {
    "step": 56,
    "epoch": 1,
    "batch_in_epoch": 55,
    "theta1": 0.005166,
    "theta2": 0.008606,
    "loss": 1.255973,
    "eta2d_x": 0.724511,
    "eta2d_y": 0.689263,
    "speed": 9.9e-05,
    "heading_rad": 0.341801,
    "drift_mean": -0.0275701,
    "drift_var": 0.0044423788,
    "opacity_var": 0.0030580468,
    "photosphere_crossed": false
  },
  {
    "step": 60,
    "epoch": 1,
    "batch_in_epoch": 59,
    "theta1": 0.005469,
    "theta2": 0.008682,
    "loss": 1.187329,
    "eta2d_x": 0.76227,
    "eta2d_y": 0.64726,
    "speed": 7.8e-05,
    "heading_rad": -0.024904,
    "drift_mean": -0.01716101,
    "drift_var": 0.0096017821,
    "opacity_var": 0.0063210359,
    "photosphere_crossed": false
  },
  {
    "step": 64,
    "epoch": 1,
    "batch_in_epoch": 63,
    "theta1": 0.005886,
    "theta2": 0.008656,
    "loss": 0.907228,
    "eta2d_x": 0.820458,
    "eta2d_y": 0.571707,
    "speed": 0.000124,
    "heading_rad": -0.043227,
    "drift_mean": -0.07002513,
    "drift_var": 0.0020284362,
    "opacity_var": 0.0063082134,
    "photosphere_crossed": false
  },
  {
    "step": 68,
    "epoch": 1,
    "batch_in_epoch": 67,
    "theta1": 0.006459,
    "theta2": 0.008834,
    "loss": 0.809386,
    "eta2d_x": 0.848248,
    "eta2d_y": 0.5296,
    "speed": 0.000124,
    "heading_rad": 0.310502,
    "drift_mean": -0.02446058,
    "drift_var": 0.0171017944,
    "opacity_var": 0.0126895475,
    "photosphere_crossed": false
  },
  {
    "step": 72,
    "epoch": 1,
    "batch_in_epoch": 71,
    "theta1": 0.006601,
    "theta2": 0.008681,
    "loss": 0.823466,
    "eta2d_x": 0.876472,
    "eta2d_y": 0.481453,
    "speed": 3.9e-05,
    "heading_rad": -2.023478,
    "drift_mean": 0.00352012,
    "drift_var": 0.0181431718,
    "opacity_var": 0.0159790259,
    "photosphere_crossed": false
  },
  {
    "step": 76,
    "epoch": 1,
    "batch_in_epoch": 75,
    "theta1": 0.00671,
    "theta2": 0.008672,
    "loss": 0.718952,
    "eta2d_x": 0.884138,
    "eta2d_y": 0.467226,
    "speed": 7.1e-05,
    "heading_rad": 0.276076,
    "drift_mean": -0.02612852,
    "drift_var": 0.0023369671,
    "opacity_var": 0.0093890385,
    "photosphere_crossed": false
  },
  {
    "step": 80,
    "epoch": 1,
    "batch_in_epoch": 79,
    "theta1": 0.006903,
    "theta2": 0.008668,
    "loss": 0.683019,
    "eta2d_x": 0.895087,
    "eta2d_y": 0.445892,
    "speed": 2.9e-05,
    "heading_rad": 0.164012,
    "drift_mean": -0.0089834,
    "drift_var": 0.0089599169,
    "opacity_var": 0.0053438391,
    "photosphere_crossed": false
  },
  {
    "step": 84,
    "epoch": 1,
    "batch_in_epoch": 83,
    "theta1": 0.00685,
    "theta2": 0.008676,
    "loss": 0.616695,
    "eta2d_x": 0.891544,
    "eta2d_y": 0.452934,
    "speed": 1.3e-05,
    "heading_rad": 2.47966,
    "drift_mean": -0.01658076,
    "drift_var": 0.0181108382,
    "opacity_var": 0.0124761216,
    "photosphere_crossed": false
  },
  {
    "step": 88,
    "epoch": 1,
    "batch_in_epoch": 87,
    "theta1": 0.006939,
    "theta2": 0.008626,
    "loss": 0.610046,
    "eta2d_x": 0.901067,
    "eta2d_y": 0.433681,
    "speed": 4.4e-05,
    "heading_rad": -0.94474,
    "drift_mean": -0.00166233,
    "drift_var": 0.0081502813,
    "opacity_var": 0.0118746067,
    "photosphere_crossed": false
  },
  {
    "step": 92,
    "epoch": 1,
    "batch_in_epoch": 91,
    "theta1": 0.006902,
    "theta2": 0.008513,
    "loss": 0.569516,
    "eta2d_x": 0.910543,
    "eta2d_y": 0.413414,
    "speed": 1.5e-05,
    "heading_rad": -0.994688,
    "drift_mean": -0.01013266,
    "drift_var": 0.0042414359,
    "opacity_var": 0.0070737597,
    "photosphere_crossed": false
  },
  {
    "step": 96,
    "epoch": 1,
    "batch_in_epoch": 95,
    "theta1": 0.0068,
    "theta2": 0.008421,
    "loss": 0.333478,
    "eta2d_x": 0.915024,
    "eta2d_y": 0.4034,
    "speed": 3.8e-05,
    "heading_rad": -2.898831,
    "drift_mean": -0.05900933,
    "drift_var": 0.02585689,
    "opacity_var": 0.0156862238,
    "photosphere_crossed": false
  },
  {
    "step": 100,
    "epoch": 1,
    "batch_in_epoch": 99,
    "theta1": 0.00677,
    "theta2": 0.008331,
    "loss": 0.412808,
    "eta2d_x": 0.922671,
    "eta2d_y": 0.385589,
    "speed": 2e-05,
    "heading_rad": -1.935002,
    "drift_mean": 0.01983249,
    "drift_var": 0.0246206496,
    "opacity_var": 0.0250554283,
    "photosphere_crossed": false
  },
  {
    "step": 104,
    "epoch": 1,
    "batch_in_epoch": 103,
    "theta1": 0.006746,
    "theta2": 0.00826,
    "loss": 0.473736,
    "eta2d_x": 0.928539,
    "eta2d_y": 0.371235,
    "speed": 4.3e-05,
    "heading_rad": -1.578996,
    "drift_mean": 0.01523198,
    "drift_var": 0.0231076789,
    "opacity_var": 0.0257653992,
    "photosphere_crossed": false
  },
  {
    "step": 108,
    "epoch": 1,
    "batch_in_epoch": 107,
    "theta1": 0.006741,
    "theta2": 0.008171,
    "loss": 0.31801,
    "eta2d_x": 0.936691,
    "eta2d_y": 0.350157,
    "speed": 2.9e-05,
    "heading_rad": 1.633957,
    "drift_mean": -0.03893141,
    "drift_var": 0.0113115289,
    "opacity_var": 0.0160227559,
    "photosphere_crossed": false
  },
  {
    "step": 112,
    "epoch": 1,
    "batch_in_epoch": 111,
    "theta1": 0.00682,
    "theta2": 0.008354,
    "loss": 0.35152,
    "eta2d_x": 0.92241,
    "eta2d_y": 0.386212,
    "speed": 7.7e-05,
    "heading_rad": 1.085956,
    "drift_mean": 0.00837744,
    "drift_var": 0.0233663499,
    "opacity_var": 0.0186705545,
    "photosphere_crossed": false
  },
  {
    "step": 116,
    "epoch": 1,
    "batch_in_epoch": 115,
    "theta1": 0.006801,
    "theta2": 0.008438,
    "loss": 0.534967,
    "eta2d_x": 0.913384,
    "eta2d_y": 0.407099,
    "speed": 3.6e-05,
    "heading_rad": -2.654282,
    "drift_mean": 0.04586174,
    "drift_var": 0.0052536154,
    "opacity_var": 0.0172648224,
    "photosphere_crossed": false
  },
  {
    "step": 120,
    "epoch": 1,
    "batch_in_epoch": 119,
    "theta1": 0.006628,
    "theta2": 0.00832,
    "loss": 0.435289,
    "eta2d_x": 0.917688,
    "eta2d_y": 0.397303,
    "speed": 4.7e-05,
    "heading_rad": -2.67582,
    "drift_mean": -0.02491953,
    "drift_var": 0.0131704156,
    "opacity_var": 0.0151823279,
    "photosphere_crossed": false
  },
  {
    "step": 124,
    "epoch": 1,
    "batch_in_epoch": 123,
    "theta1": 0.006678,
    "theta2": 0.008317,
    "loss": 0.455474,
    "eta2d_x": 0.920204,
    "eta2d_y": 0.391438,
    "speed": 2.6e-05,
    "heading_rad": 0.44493,
    "drift_mean": 0.00504626,
    "drift_var": 0.0250543539,
    "opacity_var": 0.0172012487,
    "photosphere_crossed": false
  },
  {
    "step": 128,
    "epoch": 1,
    "batch_in_epoch": 127,
    "theta1": 0.006749,
    "theta2": 0.008442,
    "loss": 0.323076,
    "eta2d_x": 0.910638,
    "eta2d_y": 0.413204,
    "speed": 3.6e-05,
    "heading_rad": 0.865491,
    "drift_mean": -0.03309964,
    "drift_var": 0.0210619916,
    "opacity_var": 0.0225276771,
    "photosphere_crossed": false
  },
  {
    "step": 132,
    "epoch": 1,
    "batch_in_epoch": 131,
    "theta1": 0.006822,
    "theta2": 0.008504,
    "loss": 0.720082,
    "eta2d_x": 0.907807,
    "eta2d_y": 0.419389,
    "speed": 1.4e-05,
    "heading_rad": -1.749431,
    "drift_mean": 0.0992515,
    "drift_var": 0.0212509941,
    "opacity_var": 0.0238032787,
    "photosphere_crossed": false
  },
  {
    "step": 136,
    "epoch": 1,
    "batch_in_epoch": 135,
    "theta1": 0.006907,
    "theta2": 0.008462,
    "loss": 0.5745,
    "eta2d_x": 0.915625,
    "eta2d_y": 0.402032,
    "speed": 4.6e-05,
    "heading_rad": 0.021592,
    "drift_mean": -0.03639546,
    "drift_var": 0.0427393489,
    "opacity_var": 0.0334426652,
    "photosphere_crossed": false
  },
  {
    "step": 140,
    "epoch": 1,
    "batch_in_epoch": 139,
    "theta1": 0.006932,
    "theta2": 0.00848,
    "loss": 0.269985,
    "eta2d_x": 0.915034,
    "eta2d_y": 0.403376,
    "speed": 1e-05,
    "heading_rad": 2.836648,
    "drift_mean": -0.0761288,
    "drift_var": 0.0138177798,
    "opacity_var": 0.0414458266,
    "photosphere_crossed": false
  },
  {
    "step": 144,
    "epoch": 1,
    "batch_in_epoch": 143,
    "theta1": 0.007054,
    "theta2": 0.008593,
    "loss": 0.273259,
    "eta2d_x": 0.909436,
    "eta2d_y": 0.415845,
    "speed": 5.8e-05,
    "heading_rad": 0.779923,
    "drift_mean": 0.00081854,
    "drift_var": 0.0088559583,
    "opacity_var": 0.0191440724,
    "photosphere_crossed": false
  },
  {
    "step": 148,
    "epoch": 1,
    "batch_in_epoch": 147,
    "theta1": 0.007164,
    "theta2": 0.008812,
    "loss": 0.367727,
    "eta2d_x": 0.893691,
    "eta2d_y": 0.448683,
    "speed": 8.7e-05,
    "heading_rad": 0.854536,
    "drift_mean": 0.02361695,
    "drift_var": 0.0099561718,
    "opacity_var": 0.0100696654,
    "photosphere_crossed": false
  },
  {
    "step": 152,
    "epoch": 1,
    "batch_in_epoch": 151,
    "theta1": 0.007279,
    "theta2": 0.008894,
    "loss": 0.425642,
    "eta2d_x": 0.891517,
    "eta2d_y": 0.452988,
    "speed": 2.4e-05,
    "heading_rad": 0.851432,
    "drift_mean": 0.01447885,
    "drift_var": 0.0269384199,
    "opacity_var": 0.0168017585,
    "photosphere_crossed": false
  },
  {
    "step": 156,
    "epoch": 1,
    "batch_in_epoch": 155,
    "theta1": 0.007246,
    "theta2": 0.009006,
    "loss": 0.288305,
    "eta2d_x": 0.879356,
    "eta2d_y": 0.476164,
    "speed": 2.7e-05,
    "heading_rad": 1.740802,
    "drift_mean": -0.03433414,
    "drift_var": 0.0033042463,
    "opacity_var": 0.0156182628,
    "photosphere_crossed": false
  },
  {
    "step": 160,
    "epoch": 1,
    "batch_in_epoch": 159,
    "theta1": 0.00733,
    "theta2": 0.009125,
    "loss": 0.424899,
    "eta2d_x": 0.872452,
    "eta2d_y": 0.4887,
    "speed": 5.5e-05,
    "heading_rad": 0.535696,
    "drift_mean": 0.03414834,
    "drift_var": 0.0171618801,
    "opacity_var": 0.0103740915,
    "photosphere_crossed": false
  },
  {
    "step": 164,
    "epoch": 1,
    "batch_in_epoch": 163,
    "theta1": 0.007583,
    "theta2": 0.009295,
    "loss": 0.264446,
    "eta2d_x": 0.869867,
    "eta2d_y": 0.493286,
    "speed": 4.8e-05,
    "heading_rad": 0.555011,
    "drift_mean": -0.04011305,
    "drift_var": 0.0142878854,
    "opacity_var": 0.0152103342,
    "photosphere_crossed": false
  },
  {
    "step": 168,
    "epoch": 1,
    "batch_in_epoch": 167,
    "theta1": 0.00768,
    "theta2": 0.009411,
    "loss": 0.336184,
    "eta2d_x": 0.864577,
    "eta2d_y": 0.502501,
    "speed": 1.7e-05,
    "heading_rad": -1.707568,
    "drift_mean": 0.01793431,
    "drift_var": 0.0161567084,
    "opacity_var": 0.0150177824,
    "photosphere_crossed": false
  },
  {
    "step": 172,
    "epoch": 1,
    "batch_in_epoch": 171,
    "theta1": 0.007633,
    "theta2": 0.00943,
    "loss": 0.444385,
    "eta2d_x": 0.86043,
    "eta2d_y": 0.509569,
    "speed": 5.3e-05,
    "heading_rad": 1.311939,
    "drift_mean": 0.02705035,
    "drift_var": 0.0143213381,
    "opacity_var": 0.0137244664,
    "photosphere_crossed": false
  },
  {
    "step": 176,
    "epoch": 1,
    "batch_in_epoch": 175,
    "theta1": 0.007727,
    "theta2": 0.00955,
    "loss": 0.457827,
    "eta2d_x": 0.854796,
    "eta2d_y": 0.518964,
    "speed": 2.7e-05,
    "heading_rad": 0.627904,
    "drift_mean": 0.00336036,
    "drift_var": 0.0100302406,
    "opacity_var": 0.011261699,
    "photosphere_crossed": false
  },
  {
    "step": 180,
    "epoch": 1,
    "batch_in_epoch": 179,
    "theta1": 0.007691,
    "theta2": 0.009462,
    "loss": 0.430141,
    "eta2d_x": 0.860683,
    "eta2d_y": 0.509141,
    "speed": 1.2e-05,
    "heading_rad": -2.208084,
    "drift_mean": -0.0069213,
    "drift_var": 0.0030968795,
    "opacity_var": 0.0075182714,
    "photosphere_crossed": false
  },
  {
    "step": 184,
    "epoch": 1,
    "batch_in_epoch": 183,
    "theta1": 0.007555,
    "theta2": 0.009685,
    "loss": 0.287969,
    "eta2d_x": 0.832848,
    "eta2d_y": 0.553502,
    "speed": 8.7e-05,
    "heading_rad": 2.238739,
    "drift_mean": -0.03554311,
    "drift_var": 0.0081413439,
    "opacity_var": 0.0062929063,
    "photosphere_crossed": false
  },
  {
    "step": 188,
    "epoch": 1,
    "batch_in_epoch": 187,
    "theta1": 0.007211,
    "theta2": 0.009707,
    "loss": 0.337908,
    "eta2d_x": 0.807589,
    "eta2d_y": 0.589746,
    "speed": 7.2e-05,
    "heading_rad": -2.747351,
    "drift_mean": 0.01248468,
    "drift_var": 0.0165305936,
    "opacity_var": 0.0120481289,
    "photosphere_crossed": false
  },
  {
    "step": 192,
    "epoch": 1,
    "batch_in_epoch": 191,
    "theta1": 0.007212,
    "theta2": 0.009649,
    "loss": 0.322202,
    "eta2d_x": 0.81351,
    "eta2d_y": 0.58155,
    "speed": 6.3e-05,
    "heading_rad": 0.857687,
    "drift_mean": -0.00392637,
    "drift_var": 0.0047179197,
    "opacity_var": 0.0101813677,
    "photosphere_crossed": false
  },
  {
    "step": 196,
    "epoch": 1,
    "batch_in_epoch": 195,
    "theta1": 0.007457,
    "theta2": 0.009979,
    "loss": 0.421051,
    "eta2d_x": 0.798858,
    "eta2d_y": 0.601519,
    "speed": 0.000115,
    "heading_rad": 1.157923,
    "drift_mean": 0.02471209,
    "drift_var": 0.0073043665,
    "opacity_var": 0.0060851581,
    "photosphere_crossed": false
  },
  {
    "step": 200,
    "epoch": 1,
    "batch_in_epoch": 199,
    "theta1": 0.007301,
    "theta2": 0.010251,
    "loss": 0.416467,
    "eta2d_x": 0.760939,
    "eta2d_y": 0.648823,
    "speed": 6.8e-05,
    "heading_rad": 2.431197,
    "drift_mean": -0.00114578,
    "drift_var": 0.0128741905,
    "opacity_var": 0.0091204813,
    "photosphere_crossed": false
  },
  {
    "step": 204,
    "epoch": 1,
    "batch_in_epoch": 203,
    "theta1": 0.007169,
    "theta2": 0.010324,
    "loss": 0.273266,
    "eta2d_x": 0.742636,
    "eta2d_y": 0.669695,
    "speed": 2.6e-05,
    "heading_rad": 2.358522,
    "drift_mean": -0.03580038,
    "drift_var": 0.0005302895,
    "opacity_var": 0.009396332,
    "photosphere_crossed": false
  },
  {
    "step": 208,
    "epoch": 1,
    "batch_in_epoch": 207,
    "theta1": 0.007353,
    "theta2": 0.010354,
    "loss": 0.425256,
    "eta2d_x": 0.755328,
    "eta2d_y": 0.655346,
    "speed": 8.1e-05,
    "heading_rad": 0.239919,
    "drift_mean": 0.03799751,
    "drift_var": 0.00248131,
    "opacity_var": 0.0057485208,
    "photosphere_crossed": false
  },
  {
    "step": 212,
    "epoch": 1,
    "batch_in_epoch": 211,
    "theta1": 0.007513,
    "theta2": 0.010466,
    "loss": 0.299136,
    "eta2d_x": 0.757792,
    "eta2d_y": 0.652496,
    "speed": 6.4e-05,
    "heading_rad": 1.247672,
    "drift_mean": -0.03153007,
    "drift_var": 0.0091055591,
    "opacity_var": 0.0062938053,
    "photosphere_crossed": false
  },
  {
    "step": 216,
    "epoch": 1,
    "batch_in_epoch": 215,
    "theta1": 0.007537,
    "theta2": 0.010661,
    "loss": 0.269127,
    "eta2d_x": 0.742182,
    "eta2d_y": 0.670198,
    "speed": 1.8e-05,
    "heading_rad": 2.530667,
    "drift_mean": -0.00750217,
    "drift_var": 0.0292488582,
    "opacity_var": 0.0179213858,
    "photosphere_crossed": false
  },
  {
    "step": 220,
    "epoch": 1,
    "batch_in_epoch": 219,
    "theta1": 0.007585,
    "theta2": 0.010456,
    "loss": 0.20001,
    "eta2d_x": 0.764299,
    "eta2d_y": 0.644862,
    "speed": 7.6e-05,
    "heading_rad": -1.042394,
    "drift_mean": -0.01727935,
    "drift_var": 0.0029385443,
    "opacity_var": 0.0163127851,
    "photosphere_crossed": false
  },
  {
    "step": 224,
    "epoch": 1,
    "batch_in_epoch": 223,
    "theta1": 0.007698,
    "theta2": 0.010401,
    "loss": 0.245001,
    "eta2d_x": 0.777493,
    "eta2d_y": 0.628891,
    "speed": 3.6e-05,
    "heading_rad": -0.507676,
    "drift_mean": 0.01124784,
    "drift_var": 0.0137384834,
    "opacity_var": 0.0081894351,
    "photosphere_crossed": false
  },
  {
    "step": 228,
    "epoch": 1,
    "batch_in_epoch": 227,
    "theta1": 0.007691,
    "theta2": 0.010311,
    "loss": 0.314352,
    "eta2d_x": 0.784997,
    "eta2d_y": 0.6195,
    "speed": 5.6e-05,
    "heading_rad": -2.074306,
    "drift_mean": 0.01733768,
    "drift_var": 0.000755688,
    "opacity_var": 0.0074409737,
    "photosphere_crossed": false
  },
  {
    "step": 232,
    "epoch": 1,
    "batch_in_epoch": 231,
    "theta1": 0.007557,
    "theta2": 0.010088,
    "loss": 0.325752,
    "eta2d_x": 0.795759,
    "eta2d_y": 0.605614,
    "speed": 4.8e-05,
    "heading_rad": -1.860514,
    "drift_mean": 0.00284999,
    "drift_var": 0.0073031048,
    "opacity_var": 0.0052852857,
    "photosphere_crossed": false
  },
  {
    "step": 236,
    "epoch": 1,
    "batch_in_epoch": 235,
    "theta1": 0.007593,
    "theta2": 0.010007,
    "loss": 0.267962,
    "eta2d_x": 0.805678,
    "eta2d_y": 0.592353,
    "speed": 3.3e-05,
    "heading_rad": -1.512583,
    "drift_mean": -0.0144475,
    "drift_var": 0.0027621704,
    "opacity_var": 0.0048083665,
    "photosphere_crossed": false
  },
  {
    "step": 240,
    "epoch": 1,
    "batch_in_epoch": 239,
    "theta1": 0.007446,
    "theta2": 0.009952,
    "loss": 0.46488,
    "eta2d_x": 0.800647,
    "eta2d_y": 0.599136,
    "speed": 7e-05,
    "heading_rad": -3.129255,
    "drift_mean": 0.04922955,
    "drift_var": 0.016140816,
    "opacity_var": 0.0101861949,
    "photosphere_crossed": false
  },
  {
    "step": 244,
    "epoch": 1,
    "batch_in_epoch": 243,
    "theta1": 0.007375,
    "theta2": 0.010031,
    "loss": 0.197486,
    "eta2d_x": 0.787967,
    "eta2d_y": 0.615718,
    "speed": 1.1e-05,
    "heading_rad": 1.524512,
    "drift_mean": -0.06684839,
    "drift_var": 0.0012565939,
    "opacity_var": 0.0107548974,
    "photosphere_crossed": false
  },
  {
    "step": 248,
    "epoch": 1,
    "batch_in_epoch": 247,
    "theta1": 0.007296,
    "theta2": 0.009883,
    "loss": 0.208898,
    "eta2d_x": 0.796352,
    "eta2d_y": 0.604834,
    "speed": 6.5e-05,
    "heading_rad": -2.120492,
    "drift_mean": 0.00285285,
    "drift_var": 0.0141791011,
    "opacity_var": 0.0153304015,
    "photosphere_crossed": false
  },
  {
    "step": 252,
    "epoch": 1,
    "batch_in_epoch": 251,
    "theta1": 0.007313,
    "theta2": 0.009776,
    "loss": 0.340656,
    "eta2d_x": 0.808188,
    "eta2d_y": 0.588925,
    "speed": 2.6e-05,
    "heading_rad": -0.221794,
    "drift_mean": 0.03293967,
    "drift_var": 0.0010687709,
    "opacity_var": 0.0083166368,
    "photosphere_crossed": false
  },
  {
    "step": 256,
    "epoch": 1,
    "batch_in_epoch": 255,
    "theta1": 0.007314,
    "theta2": 0.0099,
    "loss": 0.176841,
    "eta2d_x": 0.796074,
    "eta2d_y": 0.605199,
    "speed": 5.4e-05,
    "heading_rad": 1.598091,
    "drift_mean": -0.04095393,
    "drift_var": 0.0131458155,
    "opacity_var": 0.0083657529,
    "photosphere_crossed": false
  },
  {
    "step": 260,
    "epoch": 1,
    "batch_in_epoch": 259,
    "theta1": 0.007295,
    "theta2": 0.009998,
    "loss": 0.344255,
    "eta2d_x": 0.784982,
    "eta2d_y": 0.619518,
    "speed": 1.9e-05,
    "heading_rad": -1.497375,
    "drift_mean": 0.04185356,
    "drift_var": 0.0079230088,
    "opacity_var": 0.0116325844,
    "photosphere_crossed": false
  },
  {
    "step": 264,
    "epoch": 1,
    "batch_in_epoch": 263,
    "theta1": 0.007338,
    "theta2": 0.009954,
    "loss": 0.270409,
    "eta2d_x": 0.792591,
    "eta2d_y": 0.609754,
    "speed": 1.4e-05,
    "heading_rad": 1.681536,
    "drift_mean": -0.01846136,
    "drift_var": 0.003066387,
    "opacity_var": 0.0096850136,
    "photosphere_crossed": false
  },
  {
    "step": 268,
    "epoch": 1,
    "batch_in_epoch": 267,
    "theta1": 0.007185,
    "theta2": 0.009993,
    "loss": 0.180751,
    "eta2d_x": 0.776861,
    "eta2d_y": 0.629672,
    "speed": 5.4e-05,
    "heading_rad": -3.046341,
    "drift_mean": -0.0224146,
    "drift_var": 0.004006176,
    "opacity_var": 0.0032419722,
    "photosphere_crossed": false
  },
  {
    "step": 272,
    "epoch": 1,
    "batch_in_epoch": 271,
    "theta1": 0.006971,
    "theta2": 0.009964,
    "loss": 0.142849,
    "eta2d_x": 0.761513,
    "eta2d_y": 0.64815,
    "speed": 4.2e-05,
    "heading_rad": -2.517016,
    "drift_mean": -0.00947561,
    "drift_var": 0.0213072526,
    "opacity_var": 0.0119151417,
    "photosphere_crossed": false
  },
  {
    "step": 276,
    "epoch": 1,
    "batch_in_epoch": 275,
    "theta1": 0.006925,
    "theta2": 0.009935,
    "loss": 0.314287,
    "eta2d_x": 0.760522,
    "eta2d_y": 0.649312,
    "speed": 2.5e-05,
    "heading_rad": 2.80726,
    "drift_mean": 0.04285966,
    "drift_var": 0.0053493648,
    "opacity_var": 0.0143861352,
    "photosphere_crossed": false
  },
  {
    "step": 280,
    "epoch": 1,
    "batch_in_epoch": 279,
    "theta1": 0.00694,
    "theta2": 0.010152,
    "loss": 0.220962,
    "eta2d_x": 0.739073,
    "eta2d_y": 0.673625,
    "speed": 7.6e-05,
    "heading_rad": 1.553562,
    "drift_mean": -0.02333141,
    "drift_var": 0.0036705426,
    "opacity_var": 0.008635075,
    "photosphere_crossed": false
  },
  {
    "step": 284,
    "epoch": 1,
    "batch_in_epoch": 283,
    "theta1": 0.006812,
    "theta2": 0.010369,
    "loss": 0.316171,
    "eta2d_x": 0.703833,
    "eta2d_y": 0.710366,
    "speed": 6e-05,
    "heading_rad": 1.853194,
    "drift_mean": 0.02380244,
    "drift_var": 0.0014582681,
    "opacity_var": 0.0028033112,
    "photosphere_crossed": false
  },
  {
    "step": 288,
    "epoch": 1,
    "batch_in_epoch": 287,
    "theta1": 0.006814,
    "theta2": 0.010474,
    "loss": 0.276158,
    "eta2d_x": 0.69337,
    "eta2d_y": 0.720582,
    "speed": 1.7e-05,
    "heading_rad": 1.516568,
    "drift_mean": -0.01000326,
    "drift_var": 0.0130951794,
    "opacity_var": 0.0067639425,
    "photosphere_crossed": false
  },
  {
    "step": 292,
    "epoch": 1,
    "batch_in_epoch": 291,
    "theta1": 0.006949,
    "theta2": 0.010455,
    "loss": 0.322393,
    "eta2d_x": 0.709096,
    "eta2d_y": 0.705112,
    "speed": 5e-05,
    "heading_rad": -0.307845,
    "drift_mean": 0.0115586,
    "drift_var": 0.017939933,
    "opacity_var": 0.0145648033,
    "photosphere_crossed": false
  },
  {
    "step": 296,
    "epoch": 1,
    "batch_in_epoch": 295,
    "theta1": 0.007065,
    "theta2": 0.010439,
    "loss": 0.216958,
    "eta2d_x": 0.721787,
    "eta2d_y": 0.692115,
    "speed": 2.4e-05,
    "heading_rad": 0.368364,
    "drift_mean": -0.02635858,
    "drift_var": 0.0178778448,
    "opacity_var": 0.0165517508,
    "photosphere_crossed": false
  },
  {
    "step": 300,
    "epoch": 1,
    "batch_in_epoch": 299,
    "theta1": 0.007035,
    "theta2": 0.010564,
    "loss": 0.25126,
    "eta2d_x": 0.706786,
    "eta2d_y": 0.707427,
    "speed": 4.9e-05,
    "heading_rad": 1.871659,
    "drift_mean": 0.00857529,
    "drift_var": 0.0022174197,
    "opacity_var": 0.0093122124,
    "photosphere_crossed": false
  },
  {
    "step": 304,
    "epoch": 1,
    "batch_in_epoch": 303,
    "theta1": 0.00696,
    "theta2": 0.010593,
    "loss": 0.227572,
    "eta2d_x": 0.696535,
    "eta2d_y": 0.717522,
    "speed": 2.6e-05,
    "heading_rad": -2.454251,
    "drift_mean": -0.00592191,
    "drift_var": 0.0079660149,
    "opacity_var": 0.0053732291,
    "photosphere_crossed": false
  },
  {
    "step": 308,
    "epoch": 1,
    "batch_in_epoch": 307,
    "theta1": 0.006861,
    "theta2": 0.010525,
    "loss": 0.230415,
    "eta2d_x": 0.693217,
    "eta2d_y": 0.720729,
    "speed": 2.4e-05,
    "heading_rad": 2.329222,
    "drift_mean": 0.00071073,
    "drift_var": 0.0204137642,
    "opacity_var": 0.0130727035,
    "photosphere_crossed": false
  },
  {
    "step": 312,
    "epoch": 1,
    "batch_in_epoch": 311,
    "theta1": 0.006939,
    "theta2": 0.010632,
    "loss": 0.165935,
    "eta2d_x": 0.69067,
    "eta2d_y": 0.72317,
    "speed": 3e-05,
    "heading_rad": 0.646666,
    "drift_mean": -0.01611999,
    "drift_var": 0.0017930865,
    "opacity_var": 0.0100497036,
    "photosphere_crossed": false
  },
  {
    "step": 316,
    "epoch": 1,
    "batch_in_epoch": 315,
    "theta1": 0.007024,
    "theta2": 0.010591,
    "loss": 0.329007,
    "eta2d_x": 0.703111,
    "eta2d_y": 0.71108,
    "speed": 3.3e-05,
    "heading_rad": -1.307887,
    "drift_mean": 0.04076802,
    "drift_var": 3.82547e-05,
    "opacity_var": 0.0021003052,
    "photosphere_crossed": false
  },
  {
    "step": 320,
    "epoch": 1,
    "batch_in_epoch": 319,
    "theta1": 0.007106,
    "theta2": 0.010401,
    "loss": 0.266265,
    "eta2d_x": 0.729404,
    "eta2d_y": 0.684083,
    "speed": 4.5e-05,
    "heading_rad": -0.679537,
    "drift_mean": -0.0156854,
    "drift_var": 0.0035257602,
    "opacity_var": 0.0024420316,
    "photosphere_crossed": false
  },
  {
    "step": 324,
    "epoch": 1,
    "batch_in_epoch": 323,
    "theta1": 0.007271,
    "theta2": 0.010397,
    "loss": 0.277406,
    "eta2d_x": 0.744467,
    "eta2d_y": 0.667659,
    "speed": 4.9e-05,
    "heading_rad": 0.205731,
    "drift_mean": 0.00278528,
    "drift_var": 0.0074146435,
    "opacity_var": 0.0051704914,
    "photosphere_crossed": false
  },
  {
    "step": 328,
    "epoch": 1,
    "batch_in_epoch": 327,
    "theta1": 0.007318,
    "theta2": 0.010522,
    "loss": 0.247716,
    "eta2d_x": 0.736734,
    "eta2d_y": 0.676183,
    "speed": 4.1e-05,
    "heading_rad": 1.144925,
    "drift_mean": -0.00742272,
    "drift_var": 0.0109601332,
    "opacity_var": 0.0083863972,
    "photosphere_crossed": false
  },
  {
    "step": 332,
    "epoch": 1,
    "batch_in_epoch": 331,
    "theta1": 0.007364,
    "theta2": 0.010552,
    "loss": 0.269624,
    "eta2d_x": 0.737757,
    "eta2d_y": 0.675067,
    "speed": 2.6e-05,
    "heading_rad": -1.147448,
    "drift_mean": 0.00547719,
    "drift_var": 0.0108682253,
    "opacity_var": 0.0110553673,
    "photosphere_crossed": false
  },
  {
    "step": 336,
    "epoch": 1,
    "batch_in_epoch": 335,
    "theta1": 0.007332,
    "theta2": 0.010386,
    "loss": 0.25829,
    "eta2d_x": 0.750687,
    "eta2d_y": 0.660659,
    "speed": 4e-05,
    "heading_rad": -2.068063,
    "drift_mean": -0.00283366,
    "drift_var": 0.0026999399,
    "opacity_var": 0.0060477263,
    "photosphere_crossed": false
  },
  {
    "step": 340,
    "epoch": 1,
    "batch_in_epoch": 339,
    "theta1": 0.007305,
    "theta2": 0.01022,
    "loss": 0.334498,
    "eta2d_x": 0.764295,
    "eta2d_y": 0.644866,
    "speed": 5.6e-05,
    "heading_rad": -1.628218,
    "drift_mean": 0.0190521,
    "drift_var": 0.0020488772,
    "opacity_var": 0.0027923312,
    "photosphere_crossed": false
  },
  {
    "step": 344,
    "epoch": 1,
    "batch_in_epoch": 343,
    "theta1": 0.007211,
    "theta2": 0.01011,
    "loss": 0.265128,
    "eta2d_x": 0.767311,
    "eta2d_y": 0.641276,
    "speed": 3.5e-05,
    "heading_rad": -2.700577,
    "drift_mean": -0.01734252,
    "drift_var": 0.0051340066,
    "opacity_var": 0.0036421771,
    "photosphere_crossed": false
  },
  {
    "step": 348,
    "epoch": 1,
    "batch_in_epoch": 347,
    "theta1": 0.00714,
    "theta2": 0.010139,
    "loss": 0.284847,
    "eta2d_x": 0.758393,
    "eta2d_y": 0.651797,
    "speed": 2.8e-05,
    "heading_rad": 2.012535,
    "drift_mean": 0.0049298,
    "drift_var": 0.0108857987,
    "opacity_var": 0.0075253666,
    "photosphere_crossed": false
  },
  {
    "step": 352,
    "epoch": 1,
    "batch_in_epoch": 351,
    "theta1": 0.007149,
    "theta2": 0.010147,
    "loss": 0.219994,
    "eta2d_x": 0.758431,
    "eta2d_y": 0.651753,
    "speed": 1.1e-05,
    "heading_rad": -1.284694,
    "drift_mean": -0.01621323,
    "drift_var": 0.0047009906,
    "opacity_var": 0.0073761044,
    "photosphere_crossed": false
  },
  {
    "step": 356,
    "epoch": 1,
    "batch_in_epoch": 355,
    "theta1": 0.007372,
    "theta2": 0.010133,
    "loss": 0.147808,
    "eta2d_x": 0.7779,
    "eta2d_y": 0.628389,
    "speed": 6.6e-05,
    "heading_rad": 0.224168,
    "drift_mean": -0.01804647,
    "drift_var": 0.0044267977,
    "opacity_var": 0.0055723099,
    "photosphere_crossed": false
  },
  {
    "step": 360,
    "epoch": 1,
    "batch_in_epoch": 359,
    "theta1": 0.007746,
    "theta2": 0.010294,
    "loss": 0.327001,
    "eta2d_x": 0.790288,
    "eta2d_y": 0.612736,
    "speed": 9.7e-05,
    "heading_rad": 0.541763,
    "drift_mean": 0.04479809,
    "drift_var": 0.0199800372,
    "opacity_var": 0.0121979266,
    "photosphere_crossed": false
  },
  {
    "step": 364,
    "epoch": 1,
    "batch_in_epoch": 363,
    "theta1": 0.007957,
    "theta2": 0.010552,
    "loss": 0.242051,
    "eta2d_x": 0.782416,
    "eta2d_y": 0.622756,
    "speed": 5.5e-05,
    "heading_rad": 0.746169,
    "drift_mean": -0.0212375,
    "drift_var": 0.0070998933,
    "opacity_var": 0.0148875093,
    "photosphere_crossed": false
  },
  {
    "step": 368,
    "epoch": 1,
    "batch_in_epoch": 367,
    "theta1": 0.008056,
    "theta2": 0.010632,
    "loss": 0.242184,
    "eta2d_x": 0.78224,
    "eta2d_y": 0.622977,
    "speed": 1.8e-05,
    "heading_rad": 0.632518,
    "drift_mean": 3.335e-05,
    "drift_var": 0.0038342853,
    "opacity_var": 0.006396068,
    "photosphere_crossed": false
  },
  {
    "step": 372,
    "epoch": 1,
    "batch_in_epoch": 371,
    "theta1": 0.008307,
    "theta2": 0.010889,
    "loss": 0.355384,
    "eta2d_x": 0.777886,
    "eta2d_y": 0.628406,
    "speed": 0.000113,
    "heading_rad": 0.516062,
    "drift_mean": 0.02829994,
    "drift_var": 0.0045955045,
    "opacity_var": 0.0047727098,
    "photosphere_crossed": false
  },
  {
    "step": 376,
    "epoch": 1,
    "batch_in_epoch": 375,
    "theta1": 0.008646,
    "theta2": 0.01108,
    "loss": 0.248808,
    "eta2d_x": 0.783902,
    "eta2d_y": 0.620884,
    "speed": 6e-05,
    "heading_rad": 0.500185,
    "drift_mean": -0.02664386,
    "drift_var": 0.0105735691,
    "opacity_var": 0.0077656051,
    "photosphere_crossed": false
  },
  {
    "step": 380,
    "epoch": 1,
    "batch_in_epoch": 379,
    "theta1": 0.00856,
    "theta2": 0.011107,
    "loss": 0.208298,
    "eta2d_x": 0.776905,
    "eta2d_y": 0.629618,
    "speed": 4.7e-05,
    "heading_rad": 2.359303,
    "drift_mean": -0.01012765,
    "drift_var": 0.0005646687,
    "opacity_var": 0.0073530674,
    "photosphere_crossed": false
  },
  {
    "step": 384,
    "epoch": 1,
    "batch_in_epoch": 383,
    "theta1": 0.008391,
    "theta2": 0.011233,
    "loss": 0.245248,
    "eta2d_x": 0.757011,
    "eta2d_y": 0.653402,
    "speed": 5.7e-05,
    "heading_rad": 2.575305,
    "drift_mean": 0.00923755,
    "drift_var": 0.0005211609,
    "opacity_var": 0.0011019727,
    "photosphere_crossed": false
  },
  {
    "step": 388,
    "epoch": 1,
    "batch_in_epoch": 387,
    "theta1": 0.008297,
    "theta2": 0.011414,
    "loss": 0.203987,
    "eta2d_x": 0.73718,
    "eta2d_y": 0.675697,
    "speed": 4.2e-05,
    "heading_rad": 1.473396,
    "drift_mean": -0.01031535,
    "drift_var": 0.0005803443,
    "opacity_var": 0.0007880311,
    "photosphere_crossed": false
  },
  {
    "step": 392,
    "epoch": 1,
    "batch_in_epoch": 391,
    "theta1": 0.008306,
    "theta2": 0.011481,
    "loss": 0.442901,
    "eta2d_x": 0.732838,
    "eta2d_y": 0.680403,
    "speed": 3.1e-05,
    "heading_rad": 3.001226,
    "drift_mean": 0.0597287,
    "drift_var": 0.0170878193,
    "opacity_var": 0.0091141468,
    "photosphere_crossed": false
  },
  {
    "step": 396,
    "epoch": 1,
    "batch_in_epoch": 395,
    "theta1": 0.008064,
    "theta2": 0.011509,
    "loss": 0.422012,
    "eta2d_x": 0.713227,
    "eta2d_y": 0.700933,
    "speed": 7.8e-05,
    "heading_rad": 3.016581,
    "drift_mean": -0.00522225,
    "drift_var": 0.0230797296,
    "opacity_var": 0.0188002484,
    "photosphere_crossed": false
  },
  {
    "step": 400,
    "epoch": 1,
    "batch_in_epoch": 399,
    "theta1": 0.007601,
    "theta2": 0.011575,
    "loss": 0.234748,
    "eta2d_x": 0.670445,
    "eta2d_y": 0.741959,
    "speed": 0.000108,
    "heading_rad": 3.109863,
    "drift_mean": -0.04681608,
    "drift_var": 0.0154999921,
    "opacity_var": 0.0250296928,
    "photosphere_crossed": false
  },
  {
    "step": 404,
    "epoch": 1,
    "batch_in_epoch": 403,
    "theta1": 0.007367,
    "theta2": 0.011475,
    "loss": 0.174156,
    "eta2d_x": 0.657328,
    "eta2d_y": 0.753605,
    "speed": 5.2e-05,
    "heading_rad": -1.933087,
    "drift_mean": -0.0151481,
    "drift_var": 0.0022278005,
    "opacity_var": 0.0112283118,
    "photosphere_crossed": false
  },
  {
    "step": 408,
    "epoch": 1,
    "batch_in_epoch": 407,
    "theta1": 0.00748,
    "theta2": 0.011443,
    "loss": 0.171754,
    "eta2d_x": 0.670351,
    "eta2d_y": 0.742045,
    "speed": 4.9e-05,
    "heading_rad": 0.237224,
    "drift_mean": -0.00060045,
    "drift_var": 0.0036373122,
    "opacity_var": 0.002654034,
    "photosphere_crossed": false
  },
  {
    "step": 412,
    "epoch": 1,
    "batch_in_epoch": 411,
    "theta1": 0.007644,
    "theta2": 0.011531,
    "loss": 0.266589,
    "eta2d_x": 0.6777,
    "eta2d_y": 0.735338,
    "speed": 5.7e-05,
    "heading_rad": -0.05978,
    "drift_mean": 0.02370868,
    "drift_var": 0.0012720346,
    "opacity_var": 0.0024936427,
    "photosphere_crossed": false
  },
  {
    "step": 416,
    "epoch": 1,
    "batch_in_epoch": 415,
    "theta1": 0.00784,
    "theta2": 0.011547,
    "loss": 0.327157,
    "eta2d_x": 0.692785,
    "eta2d_y": 0.721145,
    "speed": 3.8e-05,
    "heading_rad": -0.145277,
    "drift_mean": 0.01514218,
    "drift_var": 0.011550891,
    "opacity_var": 0.0060556844,
    "photosphere_crossed": false
  },
  {
    "step": 420,
    "epoch": 1,
    "batch_in_epoch": 419,
    "theta1": 0.007978,
    "theta2": 0.011415,
    "loss": 0.158079,
    "eta2d_x": 0.714005,
    "eta2d_y": 0.70014,
    "speed": 3.7e-05,
    "heading_rad": -1.208885,
    "drift_mean": -0.04226955,
    "drift_var": 0.0029121057,
    "opacity_var": 0.0073439324,
    "photosphere_crossed": false
  },
  {
    "step": 424,
    "epoch": 1,
    "batch_in_epoch": 423,
    "theta1": 0.007933,
    "theta2": 0.011344,
    "loss": 0.169736,
    "eta2d_x": 0.716074,
    "eta2d_y": 0.698025,
    "speed": 1.3e-05,
    "heading_rad": -2.200403,
    "drift_mean": 0.0029141,
    "drift_var": 0.0025431664,
    "opacity_var": 0.0036377677,
    "photosphere_crossed": false
  },
  {
    "step": 428,
    "epoch": 1,
    "batch_in_epoch": 427,
    "theta1": 0.007822,
    "theta2": 0.011316,
    "loss": 0.196483,
    "eta2d_x": 0.709701,
    "eta2d_y": 0.704503,
    "speed": 3.2e-05,
    "heading_rad": 3.010091,
    "drift_mean": 0.00668684,
    "drift_var": 0.0110258366,
    "opacity_var": 0.0062877507,
    "photosphere_crossed": false
  },
  {
    "step": 432,
    "epoch": 1,
    "batch_in_epoch": 431,
    "theta1": 0.007977,
    "theta2": 0.011429,
    "loss": 0.245351,
    "eta2d_x": 0.712852,
    "eta2d_y": 0.701315,
    "speed": 8.7e-05,
    "heading_rad": 0.863939,
    "drift_mean": 0.01221693,
    "drift_var": 0.0046586344,
    "opacity_var": 0.0070744934,
    "photosphere_crossed": false
  },
  {
    "step": 436,
    "epoch": 1,
    "batch_in_epoch": 435,
    "theta1": 0.008247,
    "theta2": 0.011441,
    "loss": 0.167023,
    "eta2d_x": 0.731644,
    "eta2d_y": 0.681687,
    "speed": 7.1e-05,
    "heading_rad": -0.313159,
    "drift_mean": -0.0195819,
    "drift_var": 0.0144158462,
    "opacity_var": 0.0087085776,
    "photosphere_crossed": false
  },
  {
    "step": 440,
    "epoch": 1,
    "batch_in_epoch": 439,
    "theta1": 0.008414,
    "theta2": 0.011313,
    "loss": 0.23875,
    "eta2d_x": 0.752563,
    "eta2d_y": 0.65852,
    "speed": 2.6e-05,
    "heading_rad": -0.587167,
    "drift_mean": 0.01793183,
    "drift_var": 0.0020657515,
    "opacity_var": 0.008545978,
    "photosphere_crossed": false
  },
  {
    "step": 444,
    "epoch": 1,
    "batch_in_epoch": 443,
    "theta1": 0.008442,
    "theta2": 0.011361,
    "loss": 0.397539,
    "eta2d_x": 0.750803,
    "eta2d_y": 0.660527,
    "speed": 3.4e-05,
    "heading_rad": 1.256013,
    "drift_mean": 0.03969726,
    "drift_var": 0.0066178524,
    "opacity_var": 0.0072354663,
    "photosphere_crossed": false
  },
  {
    "step": 448,
    "epoch": 1,
    "batch_in_epoch": 447,
    "theta1": 0.008384,
    "theta2": 0.011426,
    "loss": 0.156351,
    "eta2d_x": 0.742211,
    "eta2d_y": 0.670167,
    "speed": 3.6e-05,
    "heading_rad": -3.040782,
    "drift_mean": -0.0602971,
    "drift_var": 0.0059466876,
    "opacity_var": 0.0081658541,
    "photosphere_crossed": false
  },
  {
    "step": 452,
    "epoch": 1,
    "batch_in_epoch": 451,
    "theta1": 0.008313,
    "theta2": 0.011262,
    "loss": 0.198282,
    "eta2d_x": 0.749716,
    "eta2d_y": 0.66176,
    "speed": 7e-05,
    "heading_rad": -1.37474,
    "drift_mean": 0.01048262,
    "drift_var": 0.0029289361,
    "opacity_var": 0.0092049815,
    "photosphere_crossed": false
  },
  {
    "step": 456,
    "epoch": 1,
    "batch_in_epoch": 455,
    "theta1": 0.008427,
    "theta2": 0.011155,
    "loss": 0.317784,
    "eta2d_x": 0.765104,
    "eta2d_y": 0.643906,
    "speed": 6.5e-05,
    "heading_rad": -0.138435,
    "drift_mean": 0.02987565,
    "drift_var": 0.0029483089,
    "opacity_var": 0.0032711009,
    "photosphere_crossed": false
  },
  {
    "step": 460,
    "epoch": 1,
    "batch_in_epoch": 459,
    "theta1": 0.008559,
    "theta2": 0.011352,
    "loss": 0.176198,
    "eta2d_x": 0.758926,
    "eta2d_y": 0.651177,
    "speed": 5.4e-05,
    "heading_rad": 1.198298,
    "drift_mean": -0.03539655,
    "drift_var": 0.0029347628,
    "opacity_var": 0.0035951547,
    "photosphere_crossed": false
  },
  {
    "step": 464,
    "epoch": 1,
    "batch_in_epoch": 463,
    "theta1": 0.008482,
    "theta2": 0.011418,
    "loss": 0.285653,
    "eta2d_x": 0.749181,
    "eta2d_y": 0.662366,
    "speed": 3.3e-05,
    "heading_rad": -2.705639,
    "drift_mean": 0.02736379,
    "drift_var": 0.0099331782,
    "opacity_var": 0.0066782795,
    "photosphere_crossed": false
  },
  {
    "step": 468,
    "epoch": 1,
    "batch_in_epoch": 467,
    "theta1": 0.008339,
    "theta2": 0.011309,
    "loss": 0.188476,
    "eta2d_x": 0.74792,
    "eta2d_y": 0.663788,
    "speed": 4.9e-05,
    "heading_rad": -2.357908,
    "drift_mean": -0.02429423,
    "drift_var": 0.0154892661,
    "opacity_var": 0.0119528487,
    "photosphere_crossed": false
  },
  {
    "step": 472,
    "epoch": 2,
    "batch_in_epoch": 3,
    "theta1": 0.008279,
    "theta2": 0.011159,
    "loss": 0.077817,
    "eta2d_x": 0.755291,
    "eta2d_y": 0.655389,
    "speed": 3.1e-05,
    "heading_rad": -1.506718,
    "drift_mean": -0.02766472,
    "drift_var": 0.0068132522,
    "opacity_var": 0.013603959,
    "photosphere_crossed": false
  },
  {
    "step": 476,
    "epoch": 2,
    "batch_in_epoch": 7,
    "theta1": 0.008342,
    "theta2": 0.011178,
    "loss": 0.262953,
    "eta2d_x": 0.757948,
    "eta2d_y": 0.652314,
    "speed": 1.8e-05,
    "heading_rad": 0.374251,
    "drift_mean": 0.04628395,
    "drift_var": 0.0116123725,
    "opacity_var": 0.0097357437,
    "photosphere_crossed": false
  },
  {
    "step": 480,
    "epoch": 2,
    "batch_in_epoch": 11,
    "theta1": 0.008405,
    "theta2": 0.011405,
    "loss": 0.10725,
    "eta2d_x": 0.745122,
    "eta2d_y": 0.666928,
    "speed": 6e-05,
    "heading_rad": 1.297683,
    "drift_mean": -0.03892575,
    "drift_var": 0.0002732747,
    "opacity_var": 0.0074183143,
    "photosphere_crossed": false
  },
  {
    "step": 484,
    "epoch": 2,
    "batch_in_epoch": 15,
    "theta1": 0.008516,
    "theta2": 0.011549,
    "loss": 0.233865,
    "eta2d_x": 0.741939,
    "eta2d_y": 0.670468,
    "speed": 3.2e-05,
    "heading_rad": 0.191647,
    "drift_mean": 0.03165384,
    "drift_var": 0.0093712627,
    "opacity_var": 0.0053943142,
    "photosphere_crossed": false
  },
  {
    "step": 488,
    "epoch": 2,
    "batch_in_epoch": 19,
    "theta1": 0.00861,
    "theta2": 0.011632,
    "loss": 0.153613,
    "eta2d_x": 0.742004,
    "eta2d_y": 0.670396,
    "speed": 2.3e-05,
    "heading_rad": 1.612406,
    "drift_mean": -0.02006311,
    "drift_var": 0.0012918793,
    "opacity_var": 0.0055885477,
    "photosphere_crossed": false
  },
  {
    "step": 492,
    "epoch": 2,
    "batch_in_epoch": 23,
    "theta1": 0.00851,
    "theta2": 0.011589,
    "loss": 0.163275,
    "eta2d_x": 0.738606,
    "eta2d_y": 0.674137,
    "speed": 3.9e-05,
    "heading_rad": -2.092711,
    "drift_mean": 0.0024154,
    "drift_var": 0.0023699436,
    "opacity_var": 0.0023594748,
    "photosphere_crossed": false
  },
  {
    "step": 496,
    "epoch": 2,
    "batch_in_epoch": 27,
    "theta1": 0.008561,
    "theta2": 0.011347,
    "loss": 0.190364,
    "eta2d_x": 0.759358,
    "eta2d_y": 0.650673,
    "speed": 8.3e-05,
    "heading_rad": -1.168753,
    "drift_mean": 0.00677237,
    "drift_var": 0.0098760415,
    "opacity_var": 0.0057861662,
    "photosphere_crossed": false
  },
  {
    "step": 500,
    "epoch": 2,
    "batch_in_epoch": 31,
    "theta1": 0.008571,
    "theta2": 0.011133,
    "loss": 0.222824,
    "eta2d_x": 0.775559,
    "eta2d_y": 0.631275,
    "speed": 3.2e-05,
    "heading_rad": -1.845467,
    "drift_mean": 0.00811505,
    "drift_var": 0.0061998649,
    "opacity_var": 0.0073135085,
    "photosphere_crossed": false
  },
  {
    "step": 504,
    "epoch": 2,
    "batch_in_epoch": 35,
    "theta1": 0.00862,
    "theta2": 0.011094,
    "loss": 0.145692,
    "eta2d_x": 0.781333,
    "eta2d_y": 0.624114,
    "speed": 2.8e-05,
    "heading_rad": 0.293736,
    "drift_mean": -0.01928312,
    "drift_var": 0.0006646453,
    "opacity_var": 0.0032296462,
    "photosphere_crossed": false
  },
  {
    "step": 508,
    "epoch": 2,
    "batch_in_epoch": 39,
    "theta1": 0.008795,
    "theta2": 0.011185,
    "loss": 0.136661,
    "eta2d_x": 0.784811,
    "eta2d_y": 0.619735,
    "speed": 4.5e-05,
    "heading_rad": 0.382058,
    "drift_mean": -0.00225773,
    "drift_var": 0.0017175368,
    "opacity_var": 0.0012158114,
    "photosphere_crossed": false
  },
  {
    "step": 512,
    "epoch": 2,
    "batch_in_epoch": 43,
    "theta1": 0.008821,
    "theta2": 0.011182,
    "loss": 0.127895,
    "eta2d_x": 0.786427,
    "eta2d_y": 0.617683,
    "speed": 1.7e-05,
    "heading_rad": -2.372356,
    "drift_mean": -0.00219136,
    "drift_var": 0.0022578663,
    "opacity_var": 0.0018969922,
    "photosphere_crossed": false
  },
  {
    "step": 516,
    "epoch": 2,
    "batch_in_epoch": 47,
    "theta1": 0.008783,
    "theta2": 0.011126,
    "loss": 0.091835,
    "eta2d_x": 0.788255,
    "eta2d_y": 0.615348,
    "speed": 1.2e-05,
    "heading_rad": -2.699596,
    "drift_mean": -0.00901507,
    "drift_var": 0.0025365791,
    "opacity_var": 0.0023028463,
    "photosphere_crossed": false
  },
  {
    "step": 520,
    "epoch": 2,
    "batch_in_epoch": 51,
    "theta1": 0.008734,
    "theta2": 0.011058,
    "loss": 0.172038,
    "eta2d_x": 0.79042,
    "eta2d_y": 0.612566,
    "speed": 2.6e-05,
    "heading_rad": -2.74233,
    "drift_mean": 0.02005072,
    "drift_var": 0.0081340702,
    "opacity_var": 0.0049367507,
    "photosphere_crossed": false
  },
  {
    "step": 524,
    "epoch": 2,
    "batch_in_epoch": 55,
    "theta1": 0.008582,
    "theta2": 0.010945,
    "loss": 0.182351,
    "eta2d_x": 0.790141,
    "eta2d_y": 0.612925,
    "speed": 4e-05,
    "heading_rad": -1.918957,
    "drift_mean": 0.00257813,
    "drift_var": 0.0026569491,
    "opacity_var": 0.0049082442,
    "photosphere_crossed": false
  },
  {
    "step": 528,
    "epoch": 2,
    "batch_in_epoch": 59,
    "theta1": 0.008571,
    "theta2": 0.010778,
    "loss": 0.126327,
    "eta2d_x": 0.801832,
    "eta2d_y": 0.59755,
    "speed": 5.2e-05,
    "heading_rad": -1.133423,
    "drift_mean": -0.01400578,
    "drift_var": 0.0014527655,
    "opacity_var": 0.0020336132,
    "photosphere_crossed": false
  },
  {
    "step": 532,
    "epoch": 2,
    "batch_in_epoch": 63,
    "theta1": 0.008725,
    "theta2": 0.010687,
    "loss": 0.204451,
    "eta2d_x": 0.81668,
    "eta2d_y": 0.577091,
    "speed": 2.7e-05,
    "heading_rad": -0.390971,
    "drift_mean": 0.0195309,
    "drift_var": 0.00699107,
    "opacity_var": 0.0042602092,
    "photosphere_crossed": false
  },
  {
    "step": 536,
    "epoch": 2,
    "batch_in_epoch": 67,
    "theta1": 0.008743,
    "theta2": 0.010665,
    "loss": 0.143641,
    "eta2d_x": 0.819134,
    "eta2d_y": 0.573602,
    "speed": 9e-06,
    "heading_rad": -1.110761,
    "drift_mean": -0.01520257,
    "drift_var": 0.0019770714,
    "opacity_var": 0.0045307825,
    "photosphere_crossed": false
  },
  {
    "step": 540,
    "epoch": 2,
    "batch_in_epoch": 71,
    "theta1": 0.008757,
    "theta2": 0.010539,
    "loss": 0.186118,
    "eta2d_x": 0.828867,
    "eta2d_y": 0.559445,
    "speed": 4.9e-05,
    "heading_rad": -0.994593,
    "drift_mean": 0.01061937,
    "drift_var": 0.000618436,
    "opacity_var": 0.001546184,
    "photosphere_crossed": false
  },
  {
    "step": 544,
    "epoch": 2,
    "batch_in_epoch": 75,
    "theta1": 0.008846,
    "theta2": 0.010356,
    "loss": 0.223157,
    "eta2d_x": 0.845991,
    "eta2d_y": 0.533197,
    "speed": 6.5e-05,
    "heading_rad": -0.862231,
    "drift_mean": 0.00925956,
    "drift_var": 0.0154272307,
    "opacity_var": 0.0079509199,
    "photosphere_crossed": false
  },
  {
    "step": 548,
    "epoch": 2,
    "batch_in_epoch": 79,
    "theta1": 0.009185,
    "theta2": 0.010238,
    "loss": 0.181808,
    "eta2d_x": 0.867694,
    "eta2d_y": 0.497098,
    "speed": 0.000101,
    "heading_rad": -0.221078,
    "drift_mean": -0.01033717,
    "drift_var": 0.0007005855,
    "opacity_var": 0.0072541776,
    "photosphere_crossed": false
  },
  {
    "step": 552,
    "epoch": 2,
    "batch_in_epoch": 83,
    "theta1": 0.009378,
    "theta2": 0.010149,
    "loss": 0.095685,
    "eta2d_x": 0.880035,
    "eta2d_y": 0.474908,
    "speed": 2.6e-05,
    "heading_rad": -0.798913,
    "drift_mean": -0.0215306,
    "drift_var": 0.0003726518,
    "opacity_var": 0.0008982122,
    "photosphere_crossed": false
  },
  {
    "step": 556,
    "epoch": 2,
    "batch_in_epoch": 87,
    "theta1": 0.009405,
    "theta2": 0.010071,
    "loss": 0.206307,
    "eta2d_x": 0.885682,
    "eta2d_y": 0.464292,
    "speed": 8e-06,
    "heading_rad": -1.034783,
    "drift_mean": 0.0276553,
    "drift_var": 0.0026925298,
    "opacity_var": 0.0019005206,
    "photosphere_crossed": false
  },
  {
    "step": 560,
    "epoch": 2,
    "batch_in_epoch": 91,
    "theta1": 0.009369,
    "theta2": 0.010031,
    "loss": 0.20447,
    "eta2d_x": 0.886954,
    "eta2d_y": 0.461857,
    "speed": 7e-06,
    "heading_rad": -2.160014,
    "drift_mean": -0.00045907,
    "drift_var": 0.0025799445,
    "opacity_var": 0.0026153377,
    "photosphere_crossed": false
  },
  {
    "step": 564,
    "epoch": 2,
    "batch_in_epoch": 95,
    "theta1": 0.009465,
    "theta2": 0.009945,
    "loss": 0.158907,
    "eta2d_x": 0.895001,
    "eta2d_y": 0.446065,
    "speed": 5.1e-05,
    "heading_rad": -0.505023,
    "drift_mean": -0.01139097,
    "drift_var": 0.002691376,
    "opacity_var": 0.0037331968,
    "photosphere_crossed": false
  },
  {
    "step": 568,
    "epoch": 2,
    "batch_in_epoch": 99,
    "theta1": 0.009558,
    "theta2": 0.009809,
    "loss": 0.113557,
    "eta2d_x": 0.905463,
    "eta2d_y": 0.424425,
    "speed": 2e-05,
    "heading_rad": -1.312739,
    "drift_mean": -0.01133742,
    "drift_var": 0.0057903496,
    "opacity_var": 0.0040822193,
    "photosphere_crossed": false
  },
  {
    "step": 572,
    "epoch": 2,
    "batch_in_epoch": 103,
    "theta1": 0.009434,
    "theta2": 0.009735,
    "loss": 0.222109,
    "eta2d_x": 0.906391,
    "eta2d_y": 0.422439,
    "speed": 7.7e-05,
    "heading_rad": 3.02208,
    "drift_mean": 0.027138,
    "drift_var": 0.0017823244,
    "opacity_var": 0.0047150991,
    "photosphere_crossed": false
  },
  {
    "step": 576,
    "epoch": 2,
    "batch_in_epoch": 107,
    "theta1": 0.009113,
    "theta2": 0.009937,
    "loss": 0.184722,
    "eta2d_x": 0.884467,
    "eta2d_y": 0.466602,
    "speed": 9.9e-05,
    "heading_rad": 2.396533,
    "drift_mean": -0.00934669,
    "drift_var": 0.0135047679,
    "opacity_var": 0.0077913458,
    "photosphere_crossed": false
  },
  {
    "step": 580,
    "epoch": 2,
    "batch_in_epoch": 111,
    "theta1": 0.008879,
    "theta2": 0.010168,
    "loss": 0.216336,
    "eta2d_x": 0.860472,
    "eta2d_y": 0.509497,
    "speed": 5.2e-05,
    "heading_rad": 2.496748,
    "drift_mean": 0.00790345,
    "drift_var": 0.0049111639,
    "opacity_var": 0.008266829,
    "photosphere_crossed": false
  },
  {
    "step": 584,
    "epoch": 2,
    "batch_in_epoch": 115,
    "theta1": 0.008945,
    "theta2": 0.010174,
    "loss": 0.203742,
    "eta2d_x": 0.862688,
    "eta2d_y": 0.505737,
    "speed": 5.6e-05,
    "heading_rad": 0.222927,
    "drift_mean": -0.0031484,
    "drift_var": 0.0091708426,
    "opacity_var": 0.0067086077,
    "photosphere_crossed": false
  },
  {
    "step": 588,
    "epoch": 2,
    "batch_in_epoch": 119,
    "theta1": 0.009097,
    "theta2": 0.01,
    "loss": 0.15321,
    "eta2d_x": 0.879867,
    "eta2d_y": 0.475221,
    "speed": 7.1e-05,
    "heading_rad": -1.04571,
    "drift_mean": -0.01263304,
    "drift_var": 0.0056474707,
    "opacity_var": 0.0068591347,
    "photosphere_crossed": false
  },
  {
    "step": 592,
    "epoch": 2,
    "batch_in_epoch": 123,
    "theta1": 0.009119,
    "theta2": 0.009874,
    "loss": 0.126456,
    "eta2d_x": 0.888643,
    "eta2d_y": 0.458599,
    "speed": 3e-05,
    "heading_rad": -2.300493,
    "drift_mean": -0.00668856,
    "drift_var": 0.0033494019,
    "opacity_var": 0.0052801861,
    "photosphere_crossed": false
  },
  {
    "step": 596,
    "epoch": 2,
    "batch_in_epoch": 127,
    "theta1": 0.008987,
    "theta2": 0.009891,
    "loss": 0.169231,
    "eta2d_x": 0.883095,
    "eta2d_y": 0.469194,
    "speed": 5.4e-05,
    "heading_rad": 2.734637,
    "drift_mean": 0.01069385,
    "drift_var": 0.006973444,
    "opacity_var": 0.0046625595,
    "photosphere_crossed": false
  },
  {
    "step": 600,
    "epoch": 2,
    "batch_in_epoch": 131,
    "theta1": 0.008758,
    "theta2": 0.009821,
    "loss": 0.174489,
    "eta2d_x": 0.879592,
    "eta2d_y": 0.475728,
    "speed": 7.1e-05,
    "heading_rad": -2.532112,
    "drift_mean": 0.00131447,
    "drift_var": 0.0026681607,
    "opacity_var": 0.0046760455,
    "photosphere_crossed": false
  },
  {
    "step": 604,
    "epoch": 2,
    "batch_in_epoch": 135,
    "theta1": 0.008609,
    "theta2": 0.009692,
    "loss": 0.201362,
    "eta2d_x": 0.882973,
    "eta2d_y": 0.469424,
    "speed": 1.3e-05,
    "heading_rad": -2.512432,
    "drift_mean": 0.00671831,
    "drift_var": 0.0018567152,
    "opacity_var": 0.0020351795,
    "photosphere_crossed": false
  },
  {
    "step": 608,
    "epoch": 2,
    "batch_in_epoch": 139,
    "theta1": 0.008646,
    "theta2": 0.009582,
    "loss": 0.195585,
    "eta2d_x": 0.891957,
    "eta2d_y": 0.452121,
    "speed": 3.4e-05,
    "heading_rad": -1.663669,
    "drift_mean": -0.00144443,
    "drift_var": 0.0041948536,
    "opacity_var": 0.0032174566,
    "photosphere_crossed": false
  },
  {
    "step": 612,
    "epoch": 2,
    "batch_in_epoch": 143,
    "theta1": 0.008594,
    "theta2": 0.009584,
    "loss": 0.197381,
    "eta2d_x": 0.890034,
    "eta2d_y": 0.455894,
    "speed": 9e-06,
    "heading_rad": 2.221111,
    "drift_mean": 0.00044917,
    "drift_var": 0.0015137489,
    "opacity_var": 0.0029154962,
    "photosphere_crossed": false
  },
  {
    "step": 616,
    "epoch": 2,
    "batch_in_epoch": 147,
    "theta1": 0.008547,
    "theta2": 0.009728,
    "loss": 0.174271,
    "eta2d_x": 0.878067,
    "eta2d_y": 0.478537,
    "speed": 2.6e-05,
    "heading_rad": 2.833521,
    "drift_mean": -0.00577756,
    "drift_var": 0.0019745981,
    "opacity_var": 0.0023510294,
    "photosphere_crossed": false
  },
  {
    "step": 620,
    "epoch": 2,
    "batch_in_epoch": 151,
    "theta1": 0.008489,
    "theta2": 0.009761,
    "loss": 0.095692,
    "eta2d_x": 0.873346,
    "eta2d_y": 0.487101,
    "speed": 1.3e-05,
    "heading_rad": 2.064857,
    "drift_mean": -0.01964477,
    "drift_var": 0.0009298279,
    "opacity_var": 0.0013650212,
    "photosphere_crossed": false
  },
  {
    "step": 624,
    "epoch": 2,
    "batch_in_epoch": 155,
    "theta1": 0.00839,
    "theta2": 0.009768,
    "loss": 0.123433,
    "eta2d_x": 0.86874,
    "eta2d_y": 0.495268,
    "speed": 3.8e-05,
    "heading_rad": 2.903035,
    "drift_mean": 0.00693525,
    "drift_var": 0.0021274947,
    "opacity_var": 0.00153807,
    "photosphere_crossed": false
  },
  {
    "step": 628,
    "epoch": 2,
    "batch_in_epoch": 159,
    "theta1": 0.008205,
    "theta2": 0.009831,
    "loss": 0.076605,
    "eta2d_x": 0.85557,
    "eta2d_y": 0.517687,
    "speed": 4.9e-05,
    "heading_rad": 2.809088,
    "drift_mean": -0.01170702,
    "drift_var": 0.0053439617,
    "opacity_var": 0.0038214438,
    "photosphere_crossed": false
  },
  {
    "step": 632,
    "epoch": 2,
    "batch_in_epoch": 163,
    "theta1": 0.00818,
    "theta2": 0.00991,
    "loss": 0.164696,
    "eta2d_x": 0.848112,
    "eta2d_y": 0.529817,
    "speed": 1.6e-05,
    "heading_rad": 0.757633,
    "drift_mean": 0.02202275,
    "drift_var": 0.0040792116,
    "opacity_var": 0.0050245797,
    "photosphere_crossed": false
  },
  {
    "step": 636,
    "epoch": 2,
    "batch_in_epoch": 167,
    "theta1": 0.008255,
    "theta2": 0.010053,
    "loss": 0.225735,
    "eta2d_x": 0.840522,
    "eta2d_y": 0.541778,
    "speed": 5.7e-05,
    "heading_rad": 1.959073,
    "drift_mean": 0.01525966,
    "drift_var": 0.0051346514,
    "opacity_var": 0.0058241638,
    "photosphere_crossed": false
  },
  {
    "step": 640,
    "epoch": 2,
    "batch_in_epoch": 171,
    "theta1": 0.008178,
    "theta2": 0.01017,
    "loss": 0.130513,
    "eta2d_x": 0.827175,
    "eta2d_y": 0.561944,
    "speed": 2.9e-05,
    "heading_rad": 2.359636,
    "drift_mean": -0.02380542,
    "drift_var": 0.0049047047,
    "opacity_var": 0.0057033201,
    "photosphere_crossed": false
  },
  {
    "step": 644,
    "epoch": 2,
    "batch_in_epoch": 175,
    "theta1": 0.008096,
    "theta2": 0.010203,
    "loss": 0.167037,
    "eta2d_x": 0.819918,
    "eta2d_y": 0.572481,
    "speed": 1.4e-05,
    "heading_rad": 1.304091,
    "drift_mean": 0.00913112,
    "drift_var": 0.0004610986,
    "opacity_var": 0.0042386082,
    "photosphere_crossed": false
  },
  {
    "step": 648,
    "epoch": 2,
    "batch_in_epoch": 179,
    "theta1": 0.008162,
    "theta2": 0.010206,
    "loss": 0.164959,
    "eta2d_x": 0.823351,
    "eta2d_y": 0.567532,
    "speed": 6e-06,
    "heading_rad": 0.105609,
    "drift_mean": -0.00051972,
    "drift_var": 0.0015835419,
    "opacity_var": 0.0009936714,
    "photosphere_crossed": false
  },
  {
    "step": 652,
    "epoch": 2,
    "batch_in_epoch": 183,
    "theta1": 0.00821,
    "theta2": 0.010233,
    "loss": 0.169802,
    "eta2d_x": 0.823879,
    "eta2d_y": 0.566765,
    "speed": 1.3e-05,
    "heading_rad": 1.256604,
    "drift_mean": 0.00121096,
    "drift_var": 0.0011935004,
    "opacity_var": 0.0013074452,
    "photosphere_crossed": false
  },
  {
    "step": 656,
    "epoch": 2,
    "batch_in_epoch": 187,
    "theta1": 0.008162,
    "theta2": 0.01027,
    "loss": 0.075414,
    "eta2d_x": 0.818239,
    "eta2d_y": 0.574878,
    "speed": 1.3e-05,
    "heading_rad": 2.959353,
    "drift_mean": -0.02359719,
    "drift_var": 0.001938729,
    "opacity_var": 0.0019454811,
    "photosphere_crossed": false
  },
  {
    "step": 660,
    "epoch": 2,
    "batch_in_epoch": 191,
    "theta1": 0.008105,
    "theta2": 0.010281,
    "loss": 0.104472,
    "eta2d_x": 0.814028,
    "eta2d_y": 0.580826,
    "speed": 8e-06,
    "heading_rad": -2.814554,
    "drift_mean": 0.00726469,
    "drift_var": 0.0027326982,
    "opacity_var": 0.002485124,
    "photosphere_crossed": false
  },
  {
    "step": 664,
    "epoch": 2,
    "batch_in_epoch": 195,
    "theta1": 0.008098,
    "theta2": 0.010255,
    "loss": 0.237004,
    "eta2d_x": 0.815784,
    "eta2d_y": 0.578357,
    "speed": 2.3e-05,
    "heading_rad": -1.376638,
    "drift_mean": 0.03313302,
    "drift_var": 0.0108093131,
    "opacity_var": 0.0071928382,
    "photosphere_crossed": false
  },
  {
    "step": 668,
    "epoch": 2,
    "batch_in_epoch": 199,
    "theta1": 0.008181,
    "theta2": 0.010323,
    "loss": 0.093699,
    "eta2d_x": 0.814984,
    "eta2d_y": 0.579484,
    "speed": 4.7e-05,
    "heading_rad": 0.668228,
    "drift_mean": -0.03582636,
    "drift_var": 0.0045360867,
    "opacity_var": 0.0081235541,
    "photosphere_crossed": false
  },
  {
    "step": 672,
    "epoch": 2,
    "batch_in_epoch": 203,
    "theta1": 0.008284,
    "theta2": 0.010563,
    "loss": 0.111189,
    "eta2d_x": 0.80187,
    "eta2d_y": 0.597499,
    "speed": 5.8e-05,
    "heading_rad": 1.217404,
    "drift_mean": 0.00437256,
    "drift_var": 0.0014506515,
    "opacity_var": 0.0071540344,
    "photosphere_crossed": false
  },
  {
    "step": 676,
    "epoch": 2,
    "batch_in_epoch": 207,
    "theta1": 0.008225,
    "theta2": 0.010804,
    "loss": 0.198929,
    "eta2d_x": 0.779344,
    "eta2d_y": 0.626597,
    "speed": 6.5e-05,
    "heading_rad": 1.955212,
    "drift_mean": 0.02193483,
    "drift_var": 0.0001235773,
    "opacity_var": 0.0011612045,
    "photosphere_crossed": false
  },
  {
    "step": 680,
    "epoch": 2,
    "batch_in_epoch": 211,
    "theta1": 0.008113,
    "theta2": 0.010878,
    "loss": 0.179933,
    "eta2d_x": 0.766218,
    "eta2d_y": 0.642581,
    "speed": 5.3e-05,
    "heading_rad": -2.912176,
    "drift_mean": -0.00474891,
    "drift_var": 0.0068087024,
    "opacity_var": 0.0033697297,
    "photosphere_crossed": false
  },
  {
    "step": 684,
    "epoch": 2,
    "batch_in_epoch": 215,
    "theta1": 0.007899,
    "theta2": 0.010797,
    "loss": 0.125512,
    "eta2d_x": 0.757912,
    "eta2d_y": 0.652357,
    "speed": 5.4e-05,
    "heading_rad": -2.782087,
    "drift_mean": -0.01360518,
    "drift_var": 0.0017420685,
    "opacity_var": 0.0039014337,
    "photosphere_crossed": false
  },
  {
    "step": 688,
    "epoch": 2,
    "batch_in_epoch": 219,
    "theta1": 0.007838,
    "theta2": 0.010769,
    "loss": 0.099442,
    "eta2d_x": 0.755902,
    "eta2d_y": 0.654684,
    "speed": 1.4e-05,
    "heading_rad": 0.50762,
    "drift_mean": -0.00651764,
    "drift_var": 0.0017681558,
    "opacity_var": 0.0017444095,
    "photosphere_crossed": false
  },
  {
    "step": 692,
    "epoch": 2,
    "batch_in_epoch": 223,
    "theta1": 0.008085,
    "theta2": 0.010824,
    "loss": 0.158385,
    "eta2d_x": 0.768617,
    "eta2d_y": 0.63971,
    "speed": 6.6e-05,
    "heading_rad": 0.246371,
    "drift_mean": 0.01473595,
    "drift_var": 0.0115942068,
    "opacity_var": 0.0061750838,
    "photosphere_crossed": false
  },
  {
    "step": 696,
    "epoch": 2,
    "batch_in_epoch": 227,
    "theta1": 0.008136,
    "theta2": 0.010617,
    "loss": 0.219525,
    "eta2d_x": 0.788574,
    "eta2d_y": 0.61494,
    "speed": 5.8e-05,
    "heading_rad": -1.914025,
    "drift_mean": 0.01528483,
    "drift_var": 0.0151715416,
    "opacity_var": 0.0120813605,
    "photosphere_crossed": false
  },
  {
    "step": 700,
    "epoch": 2,
    "batch_in_epoch": 231,
    "theta1": 0.008022,
    "theta2": 0.01042,
    "loss": 0.119221,
    "eta2d_x": 0.797579,
    "eta2d_y": 0.603214,
    "speed": 5.7e-05,
    "heading_rad": -2.126288,
    "drift_mean": -0.02507589,
    "drift_var": 0.0053924442,
    "opacity_var": 0.0095525952,
    "photosphere_crossed": false
  },
  {
    "step": 704,
    "epoch": 2,
    "batch_in_epoch": 235,
    "theta1": 0.007936,
    "theta2": 0.010273,
    "loss": 0.183014,
    "eta2d_x": 0.804546,
    "eta2d_y": 0.593891,
    "speed": 1.7e-05,
    "heading_rad": -1.621656,
    "drift_mean": 0.01594819,
    "drift_var": 0.0085852483,
    "opacity_var": 0.0077096924,
    "photosphere_crossed": false
  },
  {
    "step": 708,
    "epoch": 2,
    "batch_in_epoch": 239,
    "theta1": 0.00798,
    "theta2": 0.010282,
    "loss": 0.072184,
    "eta2d_x": 0.806411,
    "eta2d_y": 0.591355,
    "speed": 2.9e-05,
    "heading_rad": 0.936881,
    "drift_mean": -0.02770752,
    "drift_var": 0.0255902184,
    "opacity_var": 0.0156847448,
    "photosphere_crossed": false
  },
  {
    "step": 712,
    "epoch": 2,
    "batch_in_epoch": 243,
    "theta1": 0.008122,
    "theta2": 0.010565,
    "loss": 0.08499,
    "eta2d_x": 0.791989,
    "eta2d_y": 0.610535,
    "speed": 0.000108,
    "heading_rad": 1.211002,
    "drift_mean": 0.00320165,
    "drift_var": 0.0044897021,
    "opacity_var": 0.0139951887,
    "photosphere_crossed": false
  },
  {
    "step": 716,
    "epoch": 2,
    "batch_in_epoch": 247,
    "theta1": 0.008142,
    "theta2": 0.01091,
    "loss": 0.167857,
    "eta2d_x": 0.765576,
    "eta2d_y": 0.643345,
    "speed": 5.5e-05,
    "heading_rad": 1.936773,
    "drift_mean": 0.02071673,
    "drift_var": 0.0027208977,
    "opacity_var": 0.0043801619,
    "photosphere_crossed": false
  },
  {
    "step": 720,
    "epoch": 2,
    "batch_in_epoch": 251,
    "theta1": 0.008048,
    "theta2": 0.011059,
    "loss": 0.195248,
    "eta2d_x": 0.747276,
    "eta2d_y": 0.664513,
    "speed": 4.6e-05,
    "heading_rad": 2.562549,
    "drift_mean": 0.00684777,
    "drift_var": 0.001384388,
    "opacity_var": 0.0022393311,
    "photosphere_crossed": false
  },
  {
    "step": 724,
    "epoch": 2,
    "batch_in_epoch": 255,
    "theta1": 0.008035,
    "theta2": 0.011242,
    "loss": 0.176681,
    "eta2d_x": 0.731791,
    "eta2d_y": 0.681529,
    "speed": 3.3e-05,
    "heading_rad": 1.89208,
    "drift_mean": -0.00464186,
    "drift_var": 0.0064252709,
    "opacity_var": 0.0037553594,
    "photosphere_crossed": false
  },
  {
    "step": 728,
    "epoch": 2,
    "batch_in_epoch": 259,
    "theta1": 0.00791,
    "theta2": 0.011328,
    "loss": 0.169503,
    "eta2d_x": 0.715618,
    "eta2d_y": 0.698491,
    "speed": 2.8e-05,
    "heading_rad": 2.619327,
    "drift_mean": -0.00179437,
    "drift_var": 0.0050872626,
    "opacity_var": 0.0054071687,
    "photosphere_crossed": false
  },
  {
    "step": 732,
    "epoch": 2,
    "batch_in_epoch": 263,
    "theta1": 0.007829,
    "theta2": 0.011434,
    "loss": 0.105088,
    "eta2d_x": 0.700826,
    "eta2d_y": 0.713332,
    "speed": 3.6e-05,
    "heading_rad": 2.094821,
    "drift_mean": -0.01610391,
    "drift_var": 0.0024856674,
    "opacity_var": 0.0034613855,
    "photosphere_crossed": false
  },
  {
    "step": 736,
    "epoch": 2,
    "batch_in_epoch": 267,
    "theta1": 0.007845,
    "theta2": 0.011451,
    "loss": 0.12944,
    "eta2d_x": 0.700722,
    "eta2d_y": 0.713435,
    "speed": 1.1e-05,
    "heading_rad": -0.730459,
    "drift_mean": 0.00608794,
    "drift_var": 0.0015456382,
    "opacity_var": 0.0024143465,
    "photosphere_crossed": false
  },
  {
    "step": 740,
    "epoch": 2,
    "batch_in_epoch": 271,
    "theta1": 0.007766,
    "theta2": 0.011556,
    "loss": 0.09899,
    "eta2d_x": 0.686127,
    "eta2d_y": 0.727482,
    "speed": 4.7e-05,
    "heading_rad": 2.695043,
    "drift_mean": -0.00761235,
    "drift_var": 0.0058572717,
    "opacity_var": 0.0038207326,
    "photosphere_crossed": false
  },
  {
    "step": 744,
    "epoch": 2,
    "batch_in_epoch": 275,
    "theta1": 0.00763,
    "theta2": 0.011643,
    "loss": 0.088553,
    "eta2d_x": 0.667597,
    "eta2d_y": 0.744523,
    "speed": 5.2e-05,
    "heading_rad": 2.547152,
    "drift_mean": -0.00260929,
    "drift_var": 0.0033504625,
    "opacity_var": 0.004100031,
    "photosphere_crossed": false
  },
  {
    "step": 748,
    "epoch": 2,
    "batch_in_epoch": 279,
    "theta1": 0.007436,
    "theta2": 0.011767,
    "loss": 0.164809,
    "eta2d_x": 0.640466,
    "eta2d_y": 0.767987,
    "speed": 6.7e-05,
    "heading_rad": 2.686694,
    "drift_mean": 0.01906411,
    "drift_var": 0.0052721469,
    "opacity_var": 0.0039821973,
    "photosphere_crossed": false
  },
  {
    "step": 752,
    "epoch": 2,
    "batch_in_epoch": 283,
    "theta1": 0.007338,
    "theta2": 0.011831,
    "loss": 0.074552,
    "eta2d_x": 0.62614,
    "eta2d_y": 0.779711,
    "speed": 2.1e-05,
    "heading_rad": 2.227364,
    "drift_mean": -0.02256435,
    "drift_var": 0.0300111037,
    "opacity_var": 0.0161326158,
    "photosphere_crossed": false
  },
  {
    "step": 756,
    "epoch": 2,
    "batch_in_epoch": 287,
    "theta1": 0.007362,
    "theta2": 0.011873,
    "loss": 0.128647,
    "eta2d_x": 0.625245,
    "eta2d_y": 0.780429,
    "speed": 1.2e-05,
    "heading_rad": 0.442473,
    "drift_mean": 0.01352376,
    "drift_var": 0.0068989076,
    "opacity_var": 0.0180879815,
    "photosphere_crossed": false
  },
  {
    "step": 760,
    "epoch": 2,
    "batch_in_epoch": 291,
    "theta1": 0.007346,
    "theta2": 0.011843,
    "loss": 0.156703,
    "eta2d_x": 0.625953,
    "eta2d_y": 0.779861,
    "speed": 7e-06,
    "heading_rad": 2.775686,
    "drift_mean": 0.00701394,
    "drift_var": 0.0051316743,
    "opacity_var": 0.0067496763,
    "photosphere_crossed": false
  },
  {
    "step": 764,
    "epoch": 2,
    "batch_in_epoch": 295,
    "theta1": 0.007238,
    "theta2": 0.011922,
    "loss": 0.172565,
    "eta2d_x": 0.609308,
    "eta2d_y": 0.792934,
    "speed": 6.3e-05,
    "heading_rad": 2.078221,
    "drift_mean": 0.0039655,
    "drift_var": 0.0094866097,
    "opacity_var": 0.0082543676,
    "photosphere_crossed": false
  },
  {
    "step": 768,
    "epoch": 2,
    "batch_in_epoch": 299,
    "theta1": 0.007133,
    "theta2": 0.012112,
    "loss": 0.140627,
    "eta2d_x": 0.584289,
    "eta2d_y": 0.811546,
    "speed": 3.9e-05,
    "heading_rad": 2.102545,
    "drift_mean": -0.00798438,
    "drift_var": 0.0014752793,
    "opacity_var": 0.0049166964,
    "photosphere_crossed": false
  },
  {
    "step": 772,
    "epoch": 2,
    "batch_in_epoch": 303,
    "theta1": 0.007199,
    "theta2": 0.012156,
    "loss": 0.110224,
    "eta2d_x": 0.587885,
    "eta2d_y": 0.808944,
    "speed": 9e-06,
    "heading_rad": -0.910951,
    "drift_mean": -0.00760073,
    "drift_var": 0.0060604603,
    "opacity_var": 0.0045487151,
    "photosphere_crossed": false
  },
  {
    "step": 776,
    "epoch": 2,
    "batch_in_epoch": 307,
    "theta1": 0.007156,
    "theta2": 0.012062,
    "loss": 0.175931,
    "eta2d_x": 0.590442,
    "eta2d_y": 0.80708,
    "speed": 2.4e-05,
    "heading_rad": -1.483749,
    "drift_mean": 0.01642654,
    "drift_var": 0.0013247894,
    "opacity_var": 0.0035475986,
    "photosphere_crossed": false
  },
  {
    "step": 780,
    "epoch": 2,
    "batch_in_epoch": 311,
    "theta1": 0.00715,
    "theta2": 0.012011,
    "loss": 0.304226,
    "eta2d_x": 0.593615,
    "eta2d_y": 0.804749,
    "speed": 1.8e-05,
    "heading_rad": 2.108656,
    "drift_mean": 0.03207378,
    "drift_var": 0.0097319582,
    "opacity_var": 0.0072715215,
    "photosphere_crossed": false
  },
  {
    "step": 784,
    "epoch": 2,
    "batch_in_epoch": 315,
    "theta1": 0.007137,
    "theta2": 0.01201,
    "loss": 0.143543,
    "eta2d_x": 0.592391,
    "eta2d_y": 0.805651,
    "speed": 1e-05,
    "heading_rad": 0.966049,
    "drift_mean": -0.04017067,
    "drift_var": 0.0093837521,
    "opacity_var": 0.0102731117,
    "photosphere_crossed": false
  },
  {
    "step": 788,
    "epoch": 2,
    "batch_in_epoch": 319,
    "theta1": 0.007156,
    "theta2": 0.012028,
    "loss": 0.124613,
    "eta2d_x": 0.592961,
    "eta2d_y": 0.805231,
    "speed": 2.4e-05,
    "heading_rad": 0.762604,
    "drift_mean": -0.00473238,
    "drift_var": 0.0024178753,
    "opacity_var": 0.0103644501,
    "photosphere_crossed": false
  },
  {
    "step": 792,
    "epoch": 2,
    "batch_in_epoch": 323,
    "theta1": 0.0071,
    "theta2": 0.012117,
    "loss": 0.107516,
    "eta2d_x": 0.580644,
    "eta2d_y": 0.814157,
    "speed": 2.6e-05,
    "heading_rad": 1.690552,
    "drift_mean": -0.00427442,
    "drift_var": 0.0021420941,
    "opacity_var": 0.0020444313,
    "photosphere_crossed": false
  },
  {
    "step": 796,
    "epoch": 2,
    "batch_in_epoch": 327,
    "theta1": 0.007195,
    "theta2": 0.012144,
    "loss": 0.130316,
    "eta2d_x": 0.5883,
    "eta2d_y": 0.808643,
    "speed": 3.1e-05,
    "heading_rad": -0.189126,
    "drift_mean": 0.00570002,
    "drift_var": 0.0078901008,
    "opacity_var": 0.0048023594,
    "photosphere_crossed": false
  },
  {
    "step": 800,
    "epoch": 2,
    "batch_in_epoch": 331,
    "theta1": 0.007205,
    "theta2": 0.012211,
    "loss": 0.097108,
    "eta2d_x": 0.584402,
    "eta2d_y": 0.811464,
    "speed": 1.2e-05,
    "heading_rad": 1.557228,
    "drift_mean": -0.00830195,
    "drift_var": 0.0025711177,
    "opacity_var": 0.0052903968,
    "photosphere_crossed": false
  },
  {
    "step": 804,
    "epoch": 2,
    "batch_in_epoch": 335,
    "theta1": 0.007287,
    "theta2": 0.012177,
    "loss": 0.182979,
    "eta2d_x": 0.595063,
    "eta2d_y": 0.803679,
    "speed": 3.9e-05,
    "heading_rad": -0.634443,
    "drift_mean": 0.02146781,
    "drift_var": 0.003488785,
    "opacity_var": 0.0029442988,
    "photosphere_crossed": false
  },
  {
    "step": 808,
    "epoch": 2,
    "batch_in_epoch": 339,
    "theta1": 0.007393,
    "theta2": 0.012051,
    "loss": 0.115125,
    "eta2d_x": 0.614651,
    "eta2d_y": 0.788799,
    "speed": 6.3e-05,
    "heading_rad": -0.627126,
    "drift_mean": -0.01696356,
    "drift_var": 0.0009922287,
    "opacity_var": 0.0023848095,
    "photosphere_crossed": false
  },
  {
    "step": 812,
    "epoch": 2,
    "batch_in_epoch": 343,
    "theta1": 0.00753,
    "theta2": 0.011919,
    "loss": 0.112182,
    "eta2d_x": 0.637455,
    "eta2d_y": 0.770488,
    "speed": 2.9e-05,
    "heading_rad": -1.067918,
    "drift_mean": -0.00073579,
    "drift_var": 0.0021387968,
    "opacity_var": 0.0024933175,
    "photosphere_crossed": false
  },
  {
    "step": 816,
    "epoch": 2,
    "batch_in_epoch": 347,
    "theta1": 0.007563,
    "theta2": 0.012009,
    "loss": 0.15688,
    "eta2d_x": 0.633682,
    "eta2d_y": 0.773594,
    "speed": 4.2e-05,
    "heading_rad": 1.229075,
    "drift_mean": 0.0111746,
    "drift_var": 0.003179339,
    "opacity_var": 0.0024880217,
    "photosphere_crossed": false
  },
  {
    "step": 820,
    "epoch": 2,
    "batch_in_epoch": 351,
    "theta1": 0.007573,
    "theta2": 0.012096,
    "loss": 0.21686,
    "eta2d_x": 0.628133,
    "eta2d_y": 0.778106,
    "speed": 4.3e-05,
    "heading_rad": 1.609596,
    "drift_mean": 0.01499483,
    "drift_var": 0.0054258009,
    "opacity_var": 0.0042393516,
    "photosphere_crossed": false
  },
  {
    "step": 824,
    "epoch": 2,
    "batch_in_epoch": 355,
    "theta1": 0.007605,
    "theta2": 0.012215,
    "loss": 0.111968,
    "eta2d_x": 0.622408,
    "eta2d_y": 0.782693,
    "speed": 5e-06,
    "heading_rad": 0.747361,
    "drift_mean": -0.02622298,
    "drift_var": 0.005213305,
    "opacity_var": 0.0053136395,
    "photosphere_crossed": false
  },
  {
    "step": 828,
    "epoch": 2,
    "batch_in_epoch": 359,
    "theta1": 0.007664,
    "theta2": 0.012178,
    "loss": 0.227555,
    "eta2d_x": 0.630417,
    "eta2d_y": 0.776256,
    "speed": 1.2e-05,
    "heading_rad": -1.400967,
    "drift_mean": 0.02889675,
    "drift_var": 0.0012912453,
    "opacity_var": 0.0052761583,
    "photosphere_crossed": false
  },
  {
    "step": 832,
    "epoch": 2,
    "batch_in_epoch": 363,
    "theta1": 0.007568,
    "theta2": 0.012186,
    "loss": 0.099357,
    "eta2d_x": 0.62114,
    "eta2d_y": 0.783699,
    "speed": 2.7e-05,
    "heading_rad": 2.312902,
    "drift_mean": -0.03204934,
    "drift_var": 0.00021021,
    "opacity_var": 0.0015556593,
    "photosphere_crossed": false
  },
  {
    "step": 836,
    "epoch": 2,
    "batch_in_epoch": 367,
    "theta1": 0.007612,
    "theta2": 0.0122,
    "loss": 0.130308,
    "eta2d_x": 0.624129,
    "eta2d_y": 0.781321,
    "speed": 2.3e-05,
    "heading_rad": -2.075721,
    "drift_mean": 0.00773772,
    "drift_var": 0.0017566838,
    "opacity_var": 0.0018798357,
    "photosphere_crossed": false
  },
  {
    "step": 840,
    "epoch": 2,
    "batch_in_epoch": 371,
    "theta1": 0.007636,
    "theta2": 0.012159,
    "loss": 0.22124,
    "eta2d_x": 0.62925,
    "eta2d_y": 0.777203,
    "speed": 1.8e-05,
    "heading_rad": -0.473814,
    "drift_mean": 0.02273292,
    "drift_var": 0.0022211619,
    "opacity_var": 0.0019001507,
    "photosphere_crossed": false
  },
  {
    "step": 844,
    "epoch": 2,
    "batch_in_epoch": 375,
    "theta1": 0.007706,
    "theta2": 0.012213,
    "loss": 0.098298,
    "eta2d_x": 0.631608,
    "eta2d_y": 0.775288,
    "speed": 2.2e-05,
    "heading_rad": 0.232189,
    "drift_mean": -0.03073548,
    "drift_var": 0.0019600645,
    "opacity_var": 0.0026041961,
    "photosphere_crossed": false
  },
  {
    "step": 848,
    "epoch": 2,
    "batch_in_epoch": 379,
    "theta1": 0.007775,
    "theta2": 0.012267,
    "loss": 0.187858,
    "eta2d_x": 0.633764,
    "eta2d_y": 0.773526,
    "speed": 4.1e-05,
    "heading_rad": 0.892759,
    "drift_mean": 0.02238988,
    "drift_var": 0.003118997,
    "opacity_var": 0.0033628966,
    "photosphere_crossed": false
  },
  {
    "step": 852,
    "epoch": 2,
    "batch_in_epoch": 383,
    "theta1": 0.007951,
    "theta2": 0.012353,
    "loss": 0.096155,
    "eta2d_x": 0.642654,
    "eta2d_y": 0.766156,
    "speed": 4.3e-05,
    "heading_rad": 0.054254,
    "drift_mean": -0.02292556,
    "drift_var": 0.010510109,
    "opacity_var": 0.0065353109,
    "photosphere_crossed": false
  },
  {
    "step": 856,
    "epoch": 2,
    "batch_in_epoch": 387,
    "theta1": 0.00801,
    "theta2": 0.012436,
    "loss": 0.190813,
    "eta2d_x": 0.641726,
    "eta2d_y": 0.766934,
    "speed": 1.2e-05,
    "heading_rad": 1.435035,
    "drift_mean": 0.02366439,
    "drift_var": 0.0085225816,
    "opacity_var": 0.0097887055,
    "photosphere_crossed": false
  },
  {
    "step": 860,
    "epoch": 2,
    "batch_in_epoch": 391,
    "theta1": 0.008043,
    "theta2": 0.01246,
    "loss": 0.148589,
    "eta2d_x": 0.642849,
    "eta2d_y": 0.765993,
    "speed": 3.1e-05,
    "heading_rad": 1.224554,
    "drift_mean": -0.01055603,
    "drift_var": 0.0032210162,
    "opacity_var": 0.005612016,
    "photosphere_crossed": false
  },
  {
    "step": 864,
    "epoch": 2,
    "batch_in_epoch": 395,
    "theta1": 0.00805,
    "theta2": 0.012537,
    "loss": 0.115369,
    "eta2d_x": 0.638163,
    "eta2d_y": 0.769901,
    "speed": 4e-05,
    "heading_rad": 1.349668,
    "drift_mean": -0.00830502,
    "drift_var": 0.0005622499,
    "opacity_var": 0.002200795,
    "photosphere_crossed": false
  },
  {
    "step": 868,
    "epoch": 2,
    "batch_in_epoch": 399,
    "theta1": 0.007955,
    "theta2": 0.012699,
    "loss": 0.063387,
    "eta2d_x": 0.619509,
    "eta2d_y": 0.78499,
    "speed": 5.2e-05,
    "heading_rad": 2.155718,
    "drift_mean": -0.01299537,
    "drift_var": 0.0111532759,
    "opacity_var": 0.0061163723,
    "photosphere_crossed": false
  },
  {
    "step": 872,
    "epoch": 2,
    "batch_in_epoch": 403,
    "theta1": 0.007842,
    "theta2": 0.012875,
    "loss": 0.07932,
    "eta2d_x": 0.598384,
    "eta2d_y": 0.80121,
    "speed": 3.9e-05,
    "heading_rad": 2.046198,
    "drift_mean": 0.00398321,
    "drift_var": 0.0051992763,
    "opacity_var": 0.0073415536,
    "photosphere_crossed": false
  },
  {
    "step": 876,
    "epoch": 2,
    "batch_in_epoch": 407,
    "theta1": 0.007922,
    "theta2": 0.013027,
    "loss": 0.119772,
    "eta2d_x": 0.595586,
    "eta2d_y": 0.803291,
    "speed": 6.1e-05,
    "heading_rad": 0.889887,
    "drift_mean": 0.010113,
    "drift_var": 0.0012539328,
    "opacity_var": 0.0051178715,
    "photosphere_crossed": false
  },
  {
    "step": 880,
    "epoch": 2,
    "batch_in_epoch": 411,
    "theta1": 0.007973,
    "theta2": 0.013149,
    "loss": 0.061505,
    "eta2d_x": 0.592311,
    "eta2d_y": 0.805709,
    "speed": 2.1e-05,
    "heading_rad": 0.751122,
    "drift_mean": -0.0145667,
    "drift_var": 0.0010259891,
    "opacity_var": 0.0018701524,
    "photosphere_crossed": false
  },
  {
    "step": 884,
    "epoch": 2,
    "batch_in_epoch": 415,
    "theta1": 0.008106,
    "theta2": 0.013199,
    "loss": 0.156088,
    "eta2d_x": 0.600255,
    "eta2d_y": 0.799809,
    "speed": 4.2e-05,
    "heading_rad": 0.92668,
    "drift_mean": 0.02364562,
    "drift_var": 0.0007217271,
    "opacity_var": 0.0011403623,
    "photosphere_crossed": false
  },
  {
    "step": 888,
    "epoch": 2,
    "batch_in_epoch": 419,
    "theta1": 0.008029,
    "theta2": 0.013371,
    "loss": 0.138126,
    "eta2d_x": 0.583442,
    "eta2d_y": 0.812155,
    "speed": 7.2e-05,
    "heading_rad": 2.437393,
    "drift_mean": -0.00449038,
    "drift_var": 0.0049450318,
    "opacity_var": 0.002834397,
    "photosphere_crossed": false
  },
  {
    "step": 892,
    "epoch": 2,
    "batch_in_epoch": 423,
    "theta1": 0.007761,
    "theta2": 0.013606,
    "loss": 0.069407,
    "eta2d_x": 0.546816,
    "eta2d_y": 0.837253,
    "speed": 8e-05,
    "heading_rad": 2.471824,
    "drift_mean": -0.01717984,
    "drift_var": 0.0102799779,
    "opacity_var": 0.0071514196,
    "photosphere_crossed": false
  },
  {
    "step": 896,
    "epoch": 2,
    "batch_in_epoch": 427,
    "theta1": 0.007561,
    "theta2": 0.013653,
    "loss": 0.053436,
    "eta2d_x": 0.526248,
    "eta2d_y": 0.850331,
    "speed": 3.7e-05,
    "heading_rad": 2.984792,
    "drift_mean": -0.00399273,
    "drift_var": 0.0034216498,
    "opacity_var": 0.0061301799,
    "photosphere_crossed": false
  },
  {
    "step": 900,
    "epoch": 2,
    "batch_in_epoch": 431,
    "theta1": 0.00744,
    "theta2": 0.013612,
    "loss": 0.204207,
    "eta2d_x": 0.517399,
    "eta2d_y": 0.855744,
    "speed": 3e-05,
    "heading_rad": -2.934704,
    "drift_mean": 0.03769283,
    "drift_var": 0.0055375951,
    "opacity_var": 0.0051260175,
    "photosphere_crossed": false
  },
  {
    "step": 904,
    "epoch": 2,
    "batch_in_epoch": 435,
    "theta1": 0.007312,
    "theta2": 0.01354,
    "loss": 0.0618,
    "eta2d_x": 0.509316,
    "eta2d_y": 0.86058,
    "speed": 4.2e-05,
    "heading_rad": -2.452235,
    "drift_mean": -0.03560193,
    "drift_var": 0.0022485432,
    "opacity_var": 0.0046937993,
    "photosphere_crossed": false
  },
  {
    "step": 908,
    "epoch": 2,
    "batch_in_epoch": 439,
    "theta1": 0.007218,
    "theta2": 0.013472,
    "loss": 0.152297,
    "eta2d_x": 0.504033,
    "eta2d_y": 0.863685,
    "speed": 3.2e-05,
    "heading_rad": -2.739849,
    "drift_mean": 0.02262432,
    "drift_var": 0.0023667737,
    "opacity_var": 0.0047782266,
    "photosphere_crossed": false
  },
  {
    "step": 912,
    "epoch": 2,
    "batch_in_epoch": 443,
    "theta1": 0.007082,
    "theta2": 0.013431,
    "loss": 0.091373,
    "eta2d_x": 0.492778,
    "eta2d_y": 0.870155,
    "speed": 4.8e-05,
    "heading_rad": -2.463405,
    "drift_mean": -0.01523103,
    "drift_var": 0.0010977996,
    "opacity_var": 0.002071941,
    "photosphere_crossed": false
  },
  {
    "step": 916,
    "epoch": 2,
    "batch_in_epoch": 447,
    "theta1": 0.006903,
    "theta2": 0.013232,
    "loss": 0.103083,
    "eta2d_x": 0.485826,
    "eta2d_y": 0.874056,
    "speed": 7e-05,
    "heading_rad": -2.268285,
    "drift_mean": 0.00292766,
    "drift_var": 0.0023454364,
    "opacity_var": 0.0018875516,
    "photosphere_crossed": false
  },
  {
    "step": 920,
    "epoch": 2,
    "batch_in_epoch": 451,
    "theta1": 0.00684,
    "theta2": 0.013101,
    "loss": 0.095788,
    "eta2d_x": 0.486979,
    "eta2d_y": 0.873414,
    "speed": 3.4e-05,
    "heading_rad": -1.644215,
    "drift_mean": -0.00182379,
    "drift_var": 0.0007491815,
    "opacity_var": 0.0014219339,
    "photosphere_crossed": false
  },
  {
    "step": 924,
    "epoch": 2,
    "batch_in_epoch": 455,
    "theta1": 0.006902,
    "theta2": 0.01307,
    "loss": 0.056508,
    "eta2d_x": 0.49543,
    "eta2d_y": 0.868648,
    "speed": 2.1e-05,
    "heading_rad": -0.555698,
    "drift_mean": -0.00982009,
    "drift_var": 0.0034531371,
    "opacity_var": 0.0019211439,
    "photosphere_crossed": false
  },
  {
    "step": 928,
    "epoch": 2,
    "batch_in_epoch": 459,
    "theta1": 0.006961,
    "theta2": 0.013047,
    "loss": 0.060355,
    "eta2d_x": 0.50308,
    "eta2d_y": 0.86424,
    "speed": 2.2e-05,
    "heading_rad": -0.667919,
    "drift_mean": 0.00096181,
    "drift_var": 0.0015793822,
    "opacity_var": 0.0022879418,
    "photosphere_crossed": false
  },
  {
    "step": 932,
    "epoch": 2,
    "batch_in_epoch": 463,
    "theta1": 0.007032,
    "theta2": 0.012941,
    "loss": 0.107784,
    "eta2d_x": 0.517091,
    "eta2d_y": 0.85593,
    "speed": 4e-05,
    "heading_rad": -1.120701,
    "drift_mean": 0.01185717,
    "drift_var": 0.0074695322,
    "opacity_var": 0.0041848244,
    "photosphere_crossed": false
  },
  {
    "step": 936,
    "epoch": 2,
    "batch_in_epoch": 467,
    "theta1": 0.007059,
    "theta2": 0.012836,
    "loss": 0.153918,
    "eta2d_x": 0.526661,
    "eta2d_y": 0.850076,
    "speed": 1.8e-05,
    "heading_rad": -0.485816,
    "drift_mean": 0.01153367,
    "drift_var": 0.0020327324,
    "opacity_var": 0.0043005295,
    "photosphere_crossed": false
  },
  {
    "step": 940,
    "epoch": 3,
    "batch_in_epoch": 3,
    "theta1": 0.007001,
    "theta2": 0.012855,
    "loss": 0.099416,
    "eta2d_x": 0.519356,
    "eta2d_y": 0.854558,
    "speed": 3.2e-05,
    "heading_rad": 2.622603,
    "drift_mean": -0.01362554,
    "drift_var": 0.0012472504,
    "opacity_var": 0.0022423356,
    "photosphere_crossed": false
  },
  {
    "step": 944,
    "epoch": 3,
    "batch_in_epoch": 7,
    "theta1": 0.006954,
    "theta2": 0.012977,
    "loss": 0.08518,
    "eta2d_x": 0.506733,
    "eta2d_y": 0.862103,
    "speed": 3.7e-05,
    "heading_rad": 1.775242,
    "drift_mean": -0.00355897,
    "drift_var": 0.0011999981,
    "opacity_var": 0.0016393152,
    "photosphere_crossed": false
  },
  {
    "step": 948,
    "epoch": 3,
    "batch_in_epoch": 11,
    "theta1": 0.00702,
    "theta2": 0.01303,
    "loss": 0.127654,
    "eta2d_x": 0.510365,
    "eta2d_y": 0.859958,
    "speed": 9e-06,
    "heading_rad": 0.941731,
    "drift_mean": 0.01061837,
    "drift_var": 0.002092175,
    "opacity_var": 0.0016463406,
    "photosphere_crossed": false
  },
  {
    "step": 952,
    "epoch": 3,
    "batch_in_epoch": 15,
    "theta1": 0.007062,
    "theta2": 0.012944,
    "loss": 0.094715,
    "eta2d_x": 0.520057,
    "eta2d_y": 0.854132,
    "speed": 4.3e-05,
    "heading_rad": -1.535367,
    "drift_mean": -0.00823462,
    "drift_var": 0.002477731,
    "opacity_var": 0.0023550124,
    "photosphere_crossed": false
  },
  {
    "step": 956,
    "epoch": 3,
    "batch_in_epoch": 19,
    "theta1": 0.007027,
    "theta2": 0.012696,
    "loss": 0.085295,
    "eta2d_x": 0.532421,
    "eta2d_y": 0.84648,
    "speed": 6.7e-05,
    "heading_rad": -1.865893,
    "drift_mean": -0.00235507,
    "drift_var": 0.0037879235,
    "opacity_var": 0.0028012898,
    "photosphere_crossed": false
  },
  {
    "step": 960,
    "epoch": 3,
    "batch_in_epoch": 23,
    "theta1": 0.006938,
    "theta2": 0.012414,
    "loss": 0.131753,
    "eta2d_x": 0.541995,
    "eta2d_y": 0.840381,
    "speed": 6e-05,
    "heading_rad": -1.973447,
    "drift_mean": 0.01161436,
    "drift_var": 0.0029924296,
    "opacity_var": 0.0031313658,
    "photosphere_crossed": false
  },
  {
    "step": 964,
    "epoch": 3,
    "batch_in_epoch": 27,
    "theta1": 0.006962,
    "theta2": 0.012252,
    "loss": 0.072225,
    "eta2d_x": 0.556078,
    "eta2d_y": 0.83113,
    "speed": 4.3e-05,
    "heading_rad": -1.819531,
    "drift_mean": -0.01488195,
    "drift_var": 0.0066121894,
    "opacity_var": 0.0044268971,
    "photosphere_crossed": false
  },
  {
    "step": 968,
    "epoch": 3,
    "batch_in_epoch": 31,
    "theta1": 0.006888,
    "theta2": 0.012125,
    "loss": 0.130232,
    "eta2d_x": 0.557328,
    "eta2d_y": 0.830292,
    "speed": 2.3e-05,
    "heading_rad": -2.621883,
    "drift_mean": 0.01450179,
    "drift_var": 0.0057364841,
    "opacity_var": 0.0061444658,
    "photosphere_crossed": false
  },
  {
    "step": 972,
    "epoch": 3,
    "batch_in_epoch": 35,
    "theta1": 0.006758,
    "theta2": 0.012083,
    "loss": 0.112311,
    "eta2d_x": 0.545813,
    "eta2d_y": 0.837907,
    "speed": 5.6e-05,
    "heading_rad": -2.598425,
    "drift_mean": -0.00448031,
    "drift_var": 0.0027721392,
    "opacity_var": 0.0044850825,
    "photosphere_crossed": false
  },
  {
    "step": 976,
    "epoch": 3,
    "batch_in_epoch": 39,
    "theta1": 0.006555,
    "theta2": 0.011972,
    "loss": 0.059644,
    "eta2d_x": 0.53043,
    "eta2d_y": 0.847728,
    "speed": 5.9e-05,
    "heading_rad": -2.601084,
    "drift_mean": -0.01316676,
    "drift_var": 0.006292729,
    "opacity_var": 0.0043728119,
    "photosphere_crossed": false
  },
  {
    "step": 980,
    "epoch": 3,
    "batch_in_epoch": 43,
    "theta1": 0.006351,
    "theta2": 0.011877,
    "loss": 0.182503,
    "eta2d_x": 0.512274,
    "eta2d_y": 0.858822,
    "speed": 3.5e-05,
    "heading_rad": -2.831235,
    "drift_mean": 0.03071476,
    "drift_var": 0.0022751216,
    "opacity_var": 0.0042375542,
    "photosphere_crossed": false
  },
  {
    "step": 984,
    "epoch": 3,
    "batch_in_epoch": 47,
    "theta1": 0.006326,
    "theta2": 0.0119,
    "loss": 0.100606,
    "eta2d_x": 0.507386,
    "eta2d_y": 0.861719,
    "speed": 1.9e-05,
    "heading_rad": 0.522255,
    "drift_mean": -0.02047418,
    "drift_var": 0.0022558,
    "opacity_var": 0.0026531502,
    "photosphere_crossed": false
  },
  {
    "step": 988,
    "epoch": 3,
    "batch_in_epoch": 51,
    "theta1": 0.006472,
    "theta2": 0.01204,
    "loss": 0.09588,
    "eta2d_x": 0.515194,
    "eta2d_y": 0.857073,
    "speed": 4.9e-05,
    "heading_rad": 1.287398,
    "drift_mean": -0.00118136,
    "drift_var": 0.0043842876,
    "opacity_var": 0.0036019424,
    "photosphere_crossed": false
  },
  {
    "step": 992,
    "epoch": 3,
    "batch_in_epoch": 55,
    "theta1": 0.006439,
    "theta2": 0.012106,
    "loss": 0.140277,
    "eta2d_x": 0.506264,
    "eta2d_y": 0.862378,
    "speed": 3.6e-05,
    "heading_rad": 2.970704,
    "drift_mean": 0.01109907,
    "drift_var": 0.001278122,
    "opacity_var": 0.0025517818,
    "photosphere_crossed": false
  },
  {
    "step": 996,
    "epoch": 3,
    "batch_in_epoch": 59,
    "theta1": 0.006299,
    "theta2": 0.012093,
    "loss": 0.128664,
    "eta2d_x": 0.489704,
    "eta2d_y": 0.871889,
    "speed": 2.8e-05,
    "heading_rad": -2.904756,
    "drift_mean": -0.00290311,
    "drift_var": 0.0057834835,
    "opacity_var": 0.0034025744,
    "photosphere_crossed": false
  },
  {
    "step": 1000,
    "epoch": 3,
    "batch_in_epoch": 63,
    "theta1": 0.006271,
    "theta2": 0.012054,
    "loss": 0.164128,
    "eta2d_x": 0.488846,
    "eta2d_y": 0.87237,
    "speed": 1e-05,
    "heading_rad": 0.601111,
    "drift_mean": 0.00886604,
    "drift_var": 0.0017730917,
    "opacity_var": 0.003790111,
    "photosphere_crossed": false
  },
  {
    "step": 1004,
    "epoch": 3,
    "batch_in_epoch": 67,
    "theta1": 0.006316,
    "theta2": 0.012141,
    "loss": 0.145215,
    "eta2d_x": 0.48831,
    "eta2d_y": 0.87267,
    "speed": 1.7e-05,
    "heading_rad": 1.136129,
    "drift_mean": -0.00472841,
    "drift_var": 0.007186167,
    "opacity_var": 0.0046773958,
    "photosphere_crossed": false
  },
  {
    "step": 1008,
    "epoch": 3,
    "batch_in_epoch": 71,
    "theta1": 0.006331,
    "theta2": 0.012161,
    "loss": 0.060454,
    "eta2d_x": 0.488781,
    "eta2d_y": 0.872407,
    "speed": 1.1e-05,
    "heading_rad": 0.383373,
    "drift_mean": -0.02119029,
    "drift_var": 0.0035574185,
    "opacity_var": 0.0055283817,
    "photosphere_crossed": false
  },
  {
    "step": 1012,
    "epoch": 3,
    "batch_in_epoch": 75,
    "theta1": 0.006414,
    "theta2": 0.012194,
    "loss": 0.084764,
    "eta2d_x": 0.496861,
    "eta2d_y": 0.86783,
    "speed": 3.3e-05,
    "heading_rad": 0.186443,
    "drift_mean": 0.00607765,
    "drift_var": 0.0025523896,
    "opacity_var": 0.0039287878,
    "photosphere_crossed": false
  },
  {
    "step": 1016,
    "epoch": 3,
    "batch_in_epoch": 79,
    "theta1": 0.006467,
    "theta2": 0.012219,
    "loss": 0.065021,
    "eta2d_x": 0.501577,
    "eta2d_y": 0.865113,
    "speed": 2e-06,
    "heading_rad": -0.775047,
    "drift_mean": -0.00493593,
    "drift_var": 0.0004655751,
    "opacity_var": 0.0013688427,
    "photosphere_crossed": false
  },
  {
    "step": 1020,
    "epoch": 3,
    "batch_in_epoch": 83,
    "theta1": 0.006425,
    "theta2": 0.012231,
    "loss": 0.108196,
    "eta2d_x": 0.495726,
    "eta2d_y": 0.868479,
    "speed": 1.3e-05,
    "heading_rad": 2.496796,
    "drift_mean": 0.01079391,
    "drift_var": 0.0004156515,
    "opacity_var": 0.00047743,
    "photosphere_crossed": false
  },
  {
    "step": 1024,
    "epoch": 3,
    "batch_in_epoch": 87,
    "theta1": 0.006393,
    "theta2": 0.012179,
    "loss": 0.090259,
    "eta2d_x": 0.495338,
    "eta2d_y": 0.8687,
    "speed": 2.4e-05,
    "heading_rad": -1.459477,
    "drift_mean": -0.00448423,
    "drift_var": 0.0046995755,
    "opacity_var": 0.0024684918,
    "photosphere_crossed": false
  },
  {
    "step": 1028,
    "epoch": 3,
    "batch_in_epoch": 91,
    "theta1": 0.006389,
    "theta2": 0.012092,
    "loss": 0.129036,
    "eta2d_x": 0.501141,
    "eta2d_y": 0.865366,
    "speed": 2.2e-05,
    "heading_rad": -1.181227,
    "drift_mean": 0.0096942,
    "drift_var": 0.0049933822,
    "opacity_var": 0.0043992556,
    "photosphere_crossed": false
  },
  {
    "step": 1032,
    "epoch": 3,
    "batch_in_epoch": 95,
    "theta1": 0.006475,
    "theta2": 0.012072,
    "loss": 0.061074,
    "eta2d_x": 0.513237,
    "eta2d_y": 0.858247,
    "speed": 4.4e-05,
    "heading_rad": -0.131543,
    "drift_mean": -0.01699063,
    "drift_var": 0.0019634283,
    "opacity_var": 0.0035126636,
    "photosphere_crossed": false
  },
  {
    "step": 1036,
    "epoch": 3,
    "batch_in_epoch": 99,
    "theta1": 0.006682,
    "theta2": 0.012076,
    "loss": 0.12262,
    "eta2d_x": 0.537616,
    "eta2d_y": 0.84319,
    "speed": 5.9e-05,
    "heading_rad": -0.109359,
    "drift_mean": 0.01538663,
    "drift_var": 0.0079222491,
    "opacity_var": 0.0054223613,
    "photosphere_crossed": false
  },
  {
    "step": 1040,
    "epoch": 3,
    "batch_in_epoch": 103,
    "theta1": 0.006949,
    "theta2": 0.012062,
    "loss": 0.112823,
    "eta2d_x": 0.568735,
    "eta2d_y": 0.822521,
    "speed": 5.5e-05,
    "heading_rad": -0.20596,
    "drift_mean": -0.00244932,
    "drift_var": 0.0047817935,
    "opacity_var": 0.0062627718,
    "photosphere_crossed": false
  },
  {
    "step": 1044,
    "epoch": 3,
    "batch_in_epoch": 107,
    "theta1": 0.00718,
    "theta2": 0.011907,
    "loss": 0.076824,
    "eta2d_x": 0.604635,
    "eta2d_y": 0.796503,
    "speed": 9e-05,
    "heading_rad": -0.605681,
    "drift_mean": -0.00899982,
    "drift_var": 0.0006856095,
    "opacity_var": 0.0030259069,
    "photosphere_crossed": false
  },
  {
    "step": 1048,
    "epoch": 3,
    "batch_in_epoch": 111,
    "theta1": 0.007493,
    "theta2": 0.011637,
    "loss": 0.093992,
    "eta2d_x": 0.655897,
    "eta2d_y": 0.75485,
    "speed": 9.6e-05,
    "heading_rad": -0.620214,
    "drift_mean": 0.00429211,
    "drift_var": 0.0141316171,
    "opacity_var": 0.0066302845,
    "photosphere_crossed": false
  },
  {
    "step": 1052,
    "epoch": 3,
    "batch_in_epoch": 115,
    "theta1": 0.007752,
    "theta2": 0.011494,
    "loss": 0.076545,
    "eta2d_x": 0.689802,
    "eta2d_y": 0.723998,
    "speed": 7.5e-05,
    "heading_rad": -0.452956,
    "drift_mean": -0.00436179,
    "drift_var": 0.0076139411,
    "opacity_var": 0.0099191575,
    "photosphere_crossed": false
  },
  {
    "step": 1056,
    "epoch": 3,
    "batch_in_epoch": 119,
    "theta1": 0.00806,
    "theta2": 0.011459,
    "loss": 0.103389,
    "eta2d_x": 0.716731,
    "eta2d_y": 0.697349,
    "speed": 7.5e-05,
    "heading_rad": -0.038298,
    "drift_mean": 0.00671107,
    "drift_var": 0.0023180488,
    "opacity_var": 0.004649928,
    "photosphere_crossed": false
  },
  {
    "step": 1060,
    "epoch": 3,
    "batch_in_epoch": 123,
    "theta1": 0.008254,
    "theta2": 0.011228,
    "loss": 0.127914,
    "eta2d_x": 0.748294,
    "eta2d_y": 0.663367,
    "speed": 7.8e-05,
    "heading_rad": -1.059448,
    "drift_mean": 0.00613114,
    "drift_var": 0.0003945187,
    "opacity_var": 0.002026443,
    "photosphere_crossed": false
  },
  {
    "step": 1064,
    "epoch": 3,
    "batch_in_epoch": 127,
    "theta1": 0.008413,
    "theta2": 0.010973,
    "loss": 0.149889,
    "eta2d_x": 0.777978,
    "eta2d_y": 0.628292,
    "speed": 6.9e-05,
    "heading_rad": -0.758572,
    "drift_mean": 0.00549391,
    "drift_var": 0.0016248779,
    "opacity_var": 0.0009379069,
    "photosphere_crossed": false
  },
  {
    "step": 1068,
    "epoch": 3,
    "batch_in_epoch": 131,
    "theta1": 0.008643,
    "theta2": 0.010901,
    "loss": 0.077376,
    "eta2d_x": 0.796747,
    "eta2d_y": 0.604313,
    "speed": 7.1e-05,
    "heading_rad": -0.115101,
    "drift_mean": -0.01812836,
    "drift_var": 0.0029892652,
    "opacity_var": 0.0021773946,
    "photosphere_crossed": false
  },
  {
    "step": 1072,
    "epoch": 3,
    "batch_in_epoch": 135,
    "theta1": 0.008814,
    "theta2": 0.010895,
    "loss": 0.106485,
    "eta2d_x": 0.806388,
    "eta2d_y": 0.591387,
    "speed": 5e-05,
    "heading_rad": 0.397682,
    "drift_mean": 0.00727742,
    "drift_var": 0.0002983325,
    "opacity_var": 0.0020490268,
    "photosphere_crossed": false
  },
  {
    "step": 1076,
    "epoch": 3,
    "batch_in_epoch": 139,
    "theta1": 0.008946,
    "theta2": 0.011008,
    "loss": 0.175692,
    "eta2d_x": 0.805222,
    "eta2d_y": 0.592973,
    "speed": 5.2e-05,
    "heading_rad": 0.462339,
    "drift_mean": 0.01730157,
    "drift_var": 0.004739999,
    "opacity_var": 0.0022642032,
    "photosphere_crossed": false
  },
  {
    "step": 1080,
    "epoch": 3,
    "batch_in_epoch": 143,
    "theta1": 0.00906,
    "theta2": 0.011046,
    "loss": 0.147686,
    "eta2d_x": 0.808369,
    "eta2d_y": 0.588676,
    "speed": 3.5e-05,
    "heading_rad": 0.216859,
    "drift_mean": -0.0070015,
    "drift_var": 0.0049656565,
    "opacity_var": 0.0044724118,
    "photosphere_crossed": false
  },
  {
    "step": 1084,
    "epoch": 3,
    "batch_in_epoch": 147,
    "theta1": 0.009194,
    "theta2": 0.011082,
    "loss": 0.155513,
    "eta2d_x": 0.812429,
    "eta2d_y": 0.583061,
    "speed": 2.9e-05,
    "heading_rad": -0.182249,
    "drift_mean": 0.00195678,
    "drift_var": 0.002858442,
    "opacity_var": 0.0049763233,
    "photosphere_crossed": false
  },
  {
    "step": 1088,
    "epoch": 3,
    "batch_in_epoch": 151,
    "theta1": 0.009237,
    "theta2": 0.011013,
    "loss": 0.171226,
    "eta2d_x": 0.818992,
    "eta2d_y": 0.573806,
    "speed": 2.7e-05,
    "heading_rad": -1.200402,
    "drift_mean": 0.00392823,
    "drift_var": 0.0068985977,
    "opacity_var": 0.0051048082,
    "photosphere_crossed": false
  },
  {
    "step": 1092,
    "epoch": 3,
    "batch_in_epoch": 155,
    "theta1": 0.009287,
    "theta2": 0.01095,
    "loss": 0.137956,
    "eta2d_x": 0.82544,
    "eta2d_y": 0.56449,
    "speed": 3.6e-05,
    "heading_rad": -1.186246,
    "drift_mean": -0.0083174,
    "drift_var": 0.0051306944,
    "opacity_var": 0.0059089312,
    "photosphere_crossed": false
  },
  {
    "step": 1096,
    "epoch": 3,
    "batch_in_epoch": 159,
    "theta1": 0.009287,
    "theta2": 0.010745,
    "loss": 0.098951,
    "eta2d_x": 0.838793,
    "eta2d_y": 0.544451,
    "speed": 5.6e-05,
    "heading_rad": -1.951764,
    "drift_mean": -0.00975121,
    "drift_var": 0.0031593422,
    "opacity_var": 0.0060716415,
    "photosphere_crossed": false
  },
  {
    "step": 1100,
    "epoch": 3,
    "batch_in_epoch": 163,
    "theta1": 0.009184,
    "theta2": 0.010626,
    "loss": 0.183932,
    "eta2d_x": 0.842241,
    "eta2d_y": 0.539102,
    "speed": 2.6e-05,
    "heading_rad": -2.33748,
    "drift_mean": 0.02124518,
    "drift_var": 0.0046978782,
    "opacity_var": 0.0041210757,
    "photosphere_crossed": false
  },
  {
    "step": 1104,
    "epoch": 3,
    "batch_in_epoch": 167,
    "theta1": 0.009066,
    "theta2": 0.010541,
    "loss": 0.18644,
    "eta2d_x": 0.842891,
    "eta2d_y": 0.538084,
    "speed": 4.4e-05,
    "heading_rad": -2.143371,
    "drift_mean": 0.00062709,
    "drift_var": 0.0032219064,
    "opacity_var": 0.0040955457,
    "photosphere_crossed": false
  },
  {
    "step": 1108,
    "epoch": 3,
    "batch_in_epoch": 171,
    "theta1": 0.008922,
    "theta2": 0.01034,
    "loss": 0.066384,
    "eta2d_x": 0.850386,
    "eta2d_y": 0.526159,
    "speed": 4.2e-05,
    "heading_rad": -2.253527,
    "drift_mean": -0.03001422,
    "drift_var": 0.0008852462,
    "opacity_var": 0.0039830524,
    "photosphere_crossed": false
  },
  {
    "step": 1112,
    "epoch": 3,
    "batch_in_epoch": 175,
    "theta1": 0.008746,
    "theta2": 0.010132,
    "loss": 0.105784,
    "eta2d_x": 0.857398,
    "eta2d_y": 0.514653,
    "speed": 7.2e-05,
    "heading_rad": -2.098873,
    "drift_mean": 0.00985007,
    "drift_var": 0.0023670813,
    "opacity_var": 0.0025798556,
    "photosphere_crossed": false
  },
  {
    "step": 1116,
    "epoch": 3,
    "batch_in_epoch": 179,
    "theta1": 0.00877,
    "theta2": 0.009971,
    "loss": 0.101353,
    "eta2d_x": 0.869668,
    "eta2d_y": 0.493638,
    "speed": 3.1e-05,
    "heading_rad": -0.684093,
    "drift_mean": -0.00110763,
    "drift_var": 0.0047749891,
    "opacity_var": 0.0033160603,
    "photosphere_crossed": false
  },
  {
    "step": 1120,
    "epoch": 3,
    "batch_in_epoch": 183,
    "theta1": 0.00879,
    "theta2": 0.009991,
    "loss": 0.068545,
    "eta2d_x": 0.869139,
    "eta2d_y": 0.494567,
    "speed": 2.2e-05,
    "heading_rad": 2.133517,
    "drift_mean": -0.00820196,
    "drift_var": 0.0026027503,
    "opacity_var": 0.0033170002,
    "photosphere_crossed": false
  },
  {
    "step": 1124,
    "epoch": 3,
    "batch_in_epoch": 187,
    "theta1": 0.008641,
    "theta2": 0.01,
    "loss": 0.117763,
    "eta2d_x": 0.862338,
    "eta2d_y": 0.506333,
    "speed": 3e-05,
    "heading_rad": 2.919394,
    "drift_mean": 0.01230449,
    "drift_var": 0.006124843,
    "opacity_var": 0.0040306929,
    "photosphere_crossed": false
  },
  {
    "step": 1128,
    "epoch": 3,
    "batch_in_epoch": 191,
    "theta1": 0.008508,
    "theta2": 0.010049,
    "loss": 0.086216,
    "eta2d_x": 0.853014,
    "eta2d_y": 0.521888,
    "speed": 4.7e-05,
    "heading_rad": -2.91126,
    "drift_mean": -0.00788686,
    "drift_var": 0.001111715,
    "opacity_var": 0.0037193998,
    "photosphere_crossed": false
  },
  {
    "step": 1132,
    "epoch": 3,
    "batch_in_epoch": 195,
    "theta1": 0.008322,
    "theta2": 0.009991,
    "loss": 0.091032,
    "eta2d_x": 0.848673,
    "eta2d_y": 0.528918,
    "speed": 4.7e-05,
    "heading_rad": -2.407804,
    "drift_mean": 0.00120405,
    "drift_var": 0.0008630174,
    "opacity_var": 0.0012610219,
    "photosphere_crossed": false
  },
  {
    "step": 1136,
    "epoch": 3,
    "batch_in_epoch": 199,
    "theta1": 0.008193,
    "theta2": 0.009892,
    "loss": 0.069448,
    "eta2d_x": 0.850176,
    "eta2d_y": 0.526499,
    "speed": 3.3e-05,
    "heading_rad": -2.374065,
    "drift_mean": -0.00539605,
    "drift_var": 0.0008115603,
    "opacity_var": 0.0008969783,
    "photosphere_crossed": false
  },
  {
    "step": 1140,
    "epoch": 3,
    "batch_in_epoch": 203,
    "theta1": 0.008279,
    "theta2": 0.009948,
    "loss": 0.081647,
    "eta2d_x": 0.849902,
    "eta2d_y": 0.526941,
    "speed": 3e-05,
    "heading_rad": 0.717098,
    "drift_mean": 0.00304976,
    "drift_var": 0.0023005258,
    "opacity_var": 0.0014937061,
    "photosphere_crossed": false
  },
  {
    "step": 1144,
    "epoch": 3,
    "batch_in_epoch": 207,
    "theta1": 0.008346,
    "theta2": 0.010004,
    "loss": 0.090755,
    "eta2d_x": 0.848847,
    "eta2d_y": 0.528639,
    "speed": 3.2e-05,
    "heading_rad": 1.027996,
    "drift_mean": 0.00227709,
    "drift_var": 0.0018469249,
    "opacity_var": 0.001892655,
    "photosphere_crossed": false
  },
  {
    "step": 1148,
    "epoch": 3,
    "batch_in_epoch": 211,
    "theta1": 0.008404,
    "theta2": 0.010115,
    "loss": 0.087881,
    "eta2d_x": 0.84312,
    "eta2d_y": 0.537726,
    "speed": 1.1e-05,
    "heading_rad": 1.218725,
    "drift_mean": -0.00071855,
    "drift_var": 0.0008276438,
    "opacity_var": 0.0011913858,
    "photosphere_crossed": false
  },
  {
    "step": 1152,
    "epoch": 3,
    "batch_in_epoch": 215,
    "theta1": 0.008374,
    "theta2": 0.010207,
    "loss": 0.072153,
    "eta2d_x": 0.83457,
    "eta2d_y": 0.550902,
    "speed": 1.9e-05,
    "heading_rad": -3.056783,
    "drift_mean": -0.00393207,
    "drift_var": 0.0046356438,
    "opacity_var": 0.0028980377,
    "photosphere_crossed": false
  },
  {
    "step": 1156,
    "epoch": 3,
    "batch_in_epoch": 219,
    "theta1": 0.008247,
    "theta2": 0.010192,
    "loss": 0.089968,
    "eta2d_x": 0.829138,
    "eta2d_y": 0.559044,
    "speed": 2.8e-05,
    "heading_rad": -2.870304,
    "drift_mean": 0.00445383,
    "drift_var": 0.0036773217,
    "opacity_var": 0.0037415354,
    "photosphere_crossed": false
  },
  {
    "step": 1160,
    "epoch": 3,
    "batch_in_epoch": 223,
    "theta1": 0.008118,
    "theta2": 0.010256,
    "loss": 0.059901,
    "eta2d_x": 0.816841,
    "eta2d_y": 0.576863,
    "speed": 3.5e-05,
    "heading_rad": 2.173312,
    "drift_mean": -0.00751672,
    "drift_var": 0.0006855768,
    "opacity_var": 0.0030555149,
    "photosphere_crossed": false
  },
  {
    "step": 1164,
    "epoch": 3,
    "batch_in_epoch": 227,
    "theta1": 0.008112,
    "theta2": 0.010455,
    "loss": 0.043722,
    "eta2d_x": 0.800249,
    "eta2d_y": 0.599667,
    "speed": 4.4e-05,
    "heading_rad": 1.375028,
    "drift_mean": -0.00404494,
    "drift_var": 0.0028890344,
    "opacity_var": 0.0015987547,
    "photosphere_crossed": false
  },
  {
    "step": 1168,
    "epoch": 3,
    "batch_in_epoch": 231,
    "theta1": 0.008246,
    "theta2": 0.010627,
    "loss": 0.080081,
    "eta2d_x": 0.794662,
    "eta2d_y": 0.607052,
    "speed": 4.8e-05,
    "heading_rad": 1.102148,
    "drift_mean": 0.00908993,
    "drift_var": 0.0025377089,
    "opacity_var": 0.0026023231,
    "photosphere_crossed": false
  },
  {
    "step": 1172,
    "epoch": 3,
    "batch_in_epoch": 235,
    "theta1": 0.008224,
    "theta2": 0.010748,
    "loss": 0.140333,
    "eta2d_x": 0.783682,
    "eta2d_y": 0.621162,
    "speed": 5.1e-05,
    "heading_rad": 2.816657,
    "drift_mean": 0.01506285,
    "drift_var": 0.0020770008,
    "opacity_var": 0.0023799296,
    "photosphere_crossed": false
  },
  {
    "step": 1176,
    "epoch": 3,
    "batch_in_epoch": 239,
    "theta1": 0.007946,
    "theta2": 0.010762,
    "loss": 0.054081,
    "eta2d_x": 0.764164,
    "eta2d_y": 0.645023,
    "speed": 7.4e-05,
    "heading_rad": 3.053437,
    "drift_mean": -0.02156287,
    "drift_var": 0.0118198744,
    "opacity_var": 0.0067466013,
    "photosphere_crossed": false
  },
  {
    "step": 1180,
    "epoch": 3,
    "batch_in_epoch": 243,
    "theta1": 0.007699,
    "theta2": 0.010843,
    "loss": 0.129875,
    "eta2d_x": 0.739129,
    "eta2d_y": 0.673564,
    "speed": 6.7e-05,
    "heading_rad": 3.000334,
    "drift_mean": 0.01894852,
    "drift_var": 0.0050355486,
    "opacity_var": 0.007873524,
    "photosphere_crossed": false
  },
  {
    "step": 1184,
    "epoch": 3,
    "batch_in_epoch": 247,
    "theta1": 0.007548,
    "theta2": 0.010795,
    "loss": 0.108368,
    "eta2d_x": 0.731229,
    "eta2d_y": 0.682132,
    "speed": 3.4e-05,
    "heading_rad": -2.280905,
    "drift_mean": -0.0053769,
    "drift_var": 0.0035962092,
    "opacity_var": 0.0039684245,
    "photosphere_crossed": false
  },
  {
    "step": 1188,
    "epoch": 3,
    "batch_in_epoch": 251,
    "theta1": 0.007388,
    "theta2": 0.010596,
    "loss": 0.093127,
    "eta2d_x": 0.73581,
    "eta2d_y": 0.677188,
    "speed": 5.8e-05,
    "heading_rad": -2.062684,
    "drift_mean": -0.00381008,
    "drift_var": 0.0006825462,
    "opacity_var": 0.0026176547,
    "photosphere_crossed": false
  },
  {
    "step": 1192,
    "epoch": 3,
    "batch_in_epoch": 255,
    "theta1": 0.007192,
    "theta2": 0.010335,
    "loss": 0.102927,
    "eta2d_x": 0.743571,
    "eta2d_y": 0.668657,
    "speed": 7.3e-05,
    "heading_rad": -2.112319,
    "drift_mean": 0.00244994,
    "drift_var": 0.000456172,
    "opacity_var": 0.001115906,
    "photosphere_crossed": false
  },
  {
    "step": 1196,
    "epoch": 3,
    "batch_in_epoch": 259,
    "theta1": 0.007106,
    "theta2": 0.010058,
    "loss": 0.08868,
    "eta2d_x": 0.763722,
    "eta2d_y": 0.645546,
    "speed": 4.1e-05,
    "heading_rad": -1.492077,
    "drift_mean": -0.00356168,
    "drift_var": 0.0018630073,
    "opacity_var": 0.0010456886,
    "photosphere_crossed": false
  },
  {
    "step": 1200,
    "epoch": 3,
    "batch_in_epoch": 263,
    "theta1": 0.007078,
    "theta2": 0.010037,
    "loss": 0.0868,
    "eta2d_x": 0.763391,
    "eta2d_y": 0.645937,
    "speed": 1.6e-05,
    "heading_rad": 1.936113,
    "drift_mean": -0.00047002,
    "drift_var": 0.0015618309,
    "opacity_var": 0.0015923692,
    "photosphere_crossed": false
  },
  {
    "step": 1204,
    "epoch": 3,
    "batch_in_epoch": 267,
    "theta1": 0.007006,
    "theta2": 0.010021,
    "loss": 0.103664,
    "eta2d_x": 0.758719,
    "eta2d_y": 0.651418,
    "speed": 4.3e-05,
    "heading_rad": -2.137432,
    "drift_mean": 0.00421594,
    "drift_var": 0.0009960561,
    "opacity_var": 0.0011632296,
    "photosphere_crossed": false
  },
  {
    "step": 1208,
    "epoch": 3,
    "batch_in_epoch": 271,
    "theta1": 0.006968,
    "theta2": 0.009969,
    "loss": 0.113519,
    "eta2d_x": 0.760844,
    "eta2d_y": 0.648935,
    "speed": 2e-05,
    "heading_rad": -1.995043,
    "drift_mean": 0.00246365,
    "drift_var": 0.002780466,
    "opacity_var": 0.0017061355,
    "photosphere_crossed": false
  },
  {
    "step": 1212,
    "epoch": 3,
    "batch_in_epoch": 275,
    "theta1": 0.00693,
    "theta2": 0.010013,
    "loss": 0.116566,
    "eta2d_x": 0.752652,
    "eta2d_y": 0.658419,
    "speed": 2.7e-05,
    "heading_rad": 1.805189,
    "drift_mean": 0.00076192,
    "drift_var": 0.000434884,
    "opacity_var": 0.0016252367,
    "photosphere_crossed": false
  },
  {
    "step": 1216,
    "epoch": 3,
    "batch_in_epoch": 279,
    "theta1": 0.006985,
    "theta2": 0.01012,
    "loss": 0.100127,
    "eta2d_x": 0.746632,
    "eta2d_y": 0.665237,
    "speed": 3.6e-05,
    "heading_rad": 0.123511,
    "drift_mean": -0.00410989,
    "drift_var": 0.0027243645,
    "opacity_var": 0.0014121324,
    "photosphere_crossed": false
  },
  {
    "step": 1220,
    "epoch": 3,
    "batch_in_epoch": 283,
    "theta1": 0.007087,
    "theta2": 0.009995,
    "loss": 0.09574,
    "eta2d_x": 0.768424,
    "eta2d_y": 0.639941,
    "speed": 6.1e-05,
    "heading_rad": -1.554935,
    "drift_mean": -0.00109677,
    "drift_var": 0.0021104009,
    "opacity_var": 0.002151791,
    "photosphere_crossed": false
  },
  {
    "step": 1224,
    "epoch": 3,
    "batch_in_epoch": 287,
    "theta1": 0.007002,
    "theta2": 0.009846,
    "loss": 0.108973,
    "eta2d_x": 0.776694,
    "eta2d_y": 0.629879,
    "speed": 4.7e-05,
    "heading_rad": -2.610064,
    "drift_mean": 0.00330833,
    "drift_var": 0.0036475451,
    "opacity_var": 0.0025638266,
    "photosphere_crossed": false
  },
  {
    "step": 1228,
    "epoch": 3,
    "batch_in_epoch": 291,
    "theta1": 0.006833,
    "theta2": 0.009926,
    "loss": 0.099932,
    "eta2d_x": 0.75281,
    "eta2d_y": 0.658238,
    "speed": 6.4e-05,
    "heading_rad": 2.210412,
    "drift_mean": -0.00226025,
    "drift_var": 0.0017449485,
    "opacity_var": 0.0024437532,
    "photosphere_crossed": false
  },
  {
    "step": 1232,
    "epoch": 3,
    "batch_in_epoch": 295,
    "theta1": 0.006691,
    "theta2": 0.010107,
    "loss": 0.072945,
    "eta2d_x": 0.718942,
    "eta2d_y": 0.69507,
    "speed": 4.8e-05,
    "heading_rad": 2.062222,
    "drift_mean": -0.00674683,
    "drift_var": 0.0021391222,
    "opacity_var": 0.0017618157,
    "photosphere_crossed": false
  },
  {
    "step": 1236,
    "epoch": 3,
    "batch_in_epoch": 299,
    "theta1": 0.0067,
    "theta2": 0.010173,
    "loss": 0.153893,
    "eta2d_x": 0.712744,
    "eta2d_y": 0.701424,
    "speed": 3e-05,
    "heading_rad": 0.26302,
    "drift_mean": 0.02023697,
    "drift_var": 0.0023020339,
    "opacity_var": 0.0021391831,
    "photosphere_crossed": false
  },
  {
    "step": 1240,
    "epoch": 3,
    "batch_in_epoch": 303,
    "theta1": 0.006762,
    "theta2": 0.010133,
    "loss": 0.126616,
    "eta2d_x": 0.723551,
    "eta2d_y": 0.690271,
    "speed": 2.4e-05,
    "heading_rad": -1.865231,
    "drift_mean": -0.00681912,
    "drift_var": 0.0047707628,
    "opacity_var": 0.0033713841,
    "photosphere_crossed": false
  },
  {
    "step": 1244,
    "epoch": 3,
    "batch_in_epoch": 307,
    "theta1": 0.006726,
    "theta2": 0.01001,
    "loss": 0.108215,
    "eta2d_x": 0.733287,
    "eta2d_y": 0.679919,
    "speed": 5e-05,
    "heading_rad": -2.128747,
    "drift_mean": -0.00460037,
    "drift_var": 0.001566374,
    "opacity_var": 0.0028433236,
    "photosphere_crossed": false
  },
  {
    "step": 1248,
    "epoch": 3,
    "batch_in_epoch": 311,
    "theta1": 0.006598,
    "theta2": 0.009865,
    "loss": 0.116943,
    "eta2d_x": 0.736172,
    "eta2d_y": 0.676795,
    "speed": 6.2e-05,
    "heading_rad": -2.411427,
    "drift_mean": 0.00218198,
    "drift_var": 0.0018078515,
    "opacity_var": 0.0023043604,
    "photosphere_crossed": false
  },
  {
    "step": 1252,
    "epoch": 3,
    "batch_in_epoch": 315,
    "theta1": 0.00633,
    "theta2": 0.009774,
    "loss": 0.14381,
    "eta2d_x": 0.716963,
    "eta2d_y": 0.697112,
    "speed": 8.1e-05,
    "heading_rad": -3.053804,
    "drift_mean": 0.00671689,
    "drift_var": 0.0031958784,
    "opacity_var": 0.0022555089,
    "photosphere_crossed": false
  },
  {
    "step": 1256,
    "epoch": 3,
    "batch_in_epoch": 319,
    "theta1": 0.006096,
    "theta2": 0.009754,
    "loss": 0.128338,
    "eta2d_x": 0.689987,
    "eta2d_y": 0.723822,
    "speed": 4.2e-05,
    "heading_rad": 3.121124,
    "drift_mean": -0.00386809,
    "drift_var": 0.0008722616,
    "opacity_var": 0.0023003168,
    "photosphere_crossed": false
  },
  {
    "step": 1260,
    "epoch": 3,
    "batch_in_epoch": 323,
    "theta1": 0.006052,
    "theta2": 0.009921,
    "loss": 0.041385,
    "eta2d_x": 0.662902,
    "eta2d_y": 0.748706,
    "speed": 4.8e-05,
    "heading_rad": 1.634767,
    "drift_mean": -0.02173808,
    "drift_var": 0.0052960723,
    "opacity_var": 0.0036571352,
    "photosphere_crossed": false
  },
  {
    "step": 1264,
    "epoch": 3,
    "batch_in_epoch": 327,
    "theta1": 0.006049,
    "theta2": 0.010088,
    "loss": 0.065632,
    "eta2d_x": 0.642308,
    "eta2d_y": 0.766446,
    "speed": 4.5e-05,
    "heading_rad": 1.825342,
    "drift_mean": 0.00606155,
    "drift_var": 9.87512e-05,
    "opacity_var": 0.0025816557,
    "photosphere_crossed": false
  },
  {
    "step": 1268,
    "epoch": 3,
    "batch_in_epoch": 331,
    "theta1": 0.00589,
    "theta2": 0.010194,
    "loss": 0.063021,
    "eta2d_x": 0.606332,
    "eta2d_y": 0.795212,
    "speed": 4.5e-05,
    "heading_rad": 2.70254,
    "drift_mean": -0.00065261,
    "drift_var": 0.0079605447,
    "opacity_var": 0.0045263988,
    "photosphere_crossed": false
  },
  {
    "step": 1272,
    "epoch": 3,
    "batch_in_epoch": 335,
    "theta1": 0.005814,
    "theta2": 0.010219,
    "loss": 0.038064,
    "eta2d_x": 0.591499,
    "eta2d_y": 0.806305,
    "speed": 2e-06,
    "heading_rad": -3.127083,
    "drift_mean": -0.00623933,
    "drift_var": 0.0003310654,
    "opacity_var": 0.0036998182,
    "photosphere_crossed": false
  },
  {
    "step": 1276,
    "epoch": 3,
    "batch_in_epoch": 339,
    "theta1": 0.00589,
    "theta2": 0.010315,
    "loss": 0.072011,
    "eta2d_x": 0.59225,
    "eta2d_y": 0.805754,
    "speed": 6.9e-05,
    "heading_rad": 1.056278,
    "drift_mean": 0.00848678,
    "drift_var": 0.0040426919,
    "opacity_var": 0.0021081261,
    "photosphere_crossed": false
  },
  {
    "step": 1280,
    "epoch": 3,
    "batch_in_epoch": 343,
    "theta1": 0.005968,
    "theta2": 0.010629,
    "loss": 0.083935,
    "eta2d_x": 0.570149,
    "eta2d_y": 0.821541,
    "speed": 6.8e-05,
    "heading_rad": 1.349044,
    "drift_mean": 0.00298094,
    "drift_var": 0.0067609797,
    "opacity_var": 0.0048727968,
    "photosphere_crossed": false
  },
  {
    "step": 1284,
    "epoch": 3,
    "batch_in_epoch": 347,
    "theta1": 0.006021,
    "theta2": 0.010884,
    "loss": 0.076243,
    "eta2d_x": 0.552387,
    "eta2d_y": 0.833588,
    "speed": 4.7e-05,
    "heading_rad": 1.303703,
    "drift_mean": -0.00192292,
    "drift_var": 0.0009957087,
    "opacity_var": 0.0035046505,
    "photosphere_crossed": false
  },
  {
    "step": 1288,
    "epoch": 3,
    "batch_in_epoch": 351,
    "theta1": 0.005968,
    "theta2": 0.010904,
    "loss": 0.032915,
    "eta2d_x": 0.542565,
    "eta2d_y": 0.840014,
    "speed": 2.5e-05,
    "heading_rad": -2.746946,
    "drift_mean": -0.01083192,
    "drift_var": 0.0040664617,
    "opacity_var": 0.0029163517,
    "photosphere_crossed": false
  },
  {
    "step": 1292,
    "epoch": 3,
    "batch_in_epoch": 355,
    "theta1": 0.005899,
    "theta2": 0.010896,
    "loss": 0.060847,
    "eta2d_x": 0.533055,
    "eta2d_y": 0.84608,
    "speed": 1.5e-05,
    "heading_rad": 2.999955,
    "drift_mean": 0.00698295,
    "drift_var": 0.0025769502,
    "opacity_var": 0.0032107992,
    "photosphere_crossed": false
  },
  {
    "step": 1296,
    "epoch": 3,
    "batch_in_epoch": 359,
    "theta1": 0.005813,
    "theta2": 0.010938,
    "loss": 0.177613,
    "eta2d_x": 0.515718,
    "eta2d_y": 0.856759,
    "speed": 3.4e-05,
    "heading_rad": 1.56342,
    "drift_mean": 0.02919137,
    "drift_var": 0.0012674837,
    "opacity_var": 0.0034319559,
    "photosphere_crossed": false
  },
  {
    "step": 1300,
    "epoch": 3,
    "batch_in_epoch": 363,
    "theta1": 0.005866,
    "theta2": 0.011134,
    "loss": 0.047818,
    "eta2d_x": 0.505986,
    "eta2d_y": 0.862542,
    "speed": 7.1e-05,
    "heading_rad": 1.390766,
    "drift_mean": -0.03244873,
    "drift_var": 0.0004665209,
    "opacity_var": 0.0017746916,
    "photosphere_crossed": false
  },
  {
    "step": 1304,
    "epoch": 3,
    "batch_in_epoch": 367,
    "theta1": 0.005988,
    "theta2": 0.011277,
    "loss": 0.113033,
    "eta2d_x": 0.511535,
    "eta2d_y": 0.859263,
    "speed": 4.1e-05,
    "heading_rad": 0.169083,
    "drift_mean": 0.01630373,
    "drift_var": 0.0019669098,
    "opacity_var": 0.0021836979,
    "photosphere_crossed": false
  },
  {
    "step": 1308,
    "epoch": 3,
    "batch_in_epoch": 371,
    "theta1": 0.006001,
    "theta2": 0.011277,
    "loss": 0.135494,
    "eta2d_x": 0.513325,
    "eta2d_y": 0.858194,
    "speed": 1.9e-05,
    "heading_rad": -1.871864,
    "drift_mean": 0.00561523,
    "drift_var": 0.0006619692,
    "opacity_var": 0.0013349136,
    "photosphere_crossed": false
  },
  {
    "step": 1312,
    "epoch": 3,
    "batch_in_epoch": 375,
    "theta1": 0.006045,
    "theta2": 0.011241,
    "loss": 0.062897,
    "eta2d_x": 0.522925,
    "eta2d_y": 0.852378,
    "speed": 7e-06,
    "heading_rad": -0.2651,
    "drift_mean": -0.01814927,
    "drift_var": 0.0047245579,
    "opacity_var": 0.0025381531,
    "photosphere_crossed": false
  },
  {
    "step": 1316,
    "epoch": 3,
    "batch_in_epoch": 379,
    "theta1": 0.006021,
    "theta2": 0.011146,
    "loss": 0.161314,
    "eta2d_x": 0.527889,
    "eta2d_y": 0.849313,
    "speed": 2.8e-05,
    "heading_rad": -2.323958,
    "drift_mean": 0.02460433,
    "drift_var": 0.0016517637,
    "opacity_var": 0.0032414891,
    "photosphere_crossed": false
  },
  {
    "step": 1320,
    "epoch": 3,
    "batch_in_epoch": 383,
    "theta1": 0.005938,
    "theta2": 0.011225,
    "loss": 0.05996,
    "eta2d_x": 0.508791,
    "eta2d_y": 0.86089,
    "speed": 4.2e-05,
    "heading_rad": 1.925588,
    "drift_mean": -0.02533855,
    "drift_var": 0.0010335104,
    "opacity_var": 0.0019238075,
    "photosphere_crossed": false
  },
  {
    "step": 1324,
    "epoch": 3,
    "batch_in_epoch": 387,
    "theta1": 0.005919,
    "theta2": 0.011251,
    "loss": 0.085142,
    "eta2d_x": 0.503637,
    "eta2d_y": 0.863915,
    "speed": 1.7e-05,
    "heading_rad": -2.033066,
    "drift_mean": 0.00629557,
    "drift_var": 0.000245065,
    "opacity_var": 0.0009395077,
    "photosphere_crossed": false
  },
  {
    "step": 1328,
    "epoch": 3,
    "batch_in_epoch": 391,
    "theta1": 0.005941,
    "theta2": 0.011276,
    "loss": 0.073542,
    "eta2d_x": 0.504746,
    "eta2d_y": 0.863268,
    "speed": 2.2e-05,
    "heading_rad": 0.97329,
    "drift_mean": -0.00289993,
    "drift_var": 2.75414e-05,
    "opacity_var": 0.0001729085,
    "photosphere_crossed": true
  },
  {
    "step": 1332,
    "epoch": 3,
    "batch_in_epoch": 395,
    "theta1": 0.006032,
    "theta2": 0.011423,
    "loss": 0.141799,
    "eta2d_x": 0.50548,
    "eta2d_y": 0.862839,
    "speed": 5e-05,
    "heading_rad": 1.191867,
    "drift_mean": 0.01706421,
    "drift_var": 0.0073236377,
    "opacity_var": 0.003428553,
    "photosphere_crossed": false
  },
  {
    "step": 1336,
    "epoch": 3,
    "batch_in_epoch": 399,
    "theta1": 0.005978,
    "theta2": 0.01146,
    "loss": 0.052723,
    "eta2d_x": 0.494642,
    "eta2d_y": 0.869097,
    "speed": 2.6e-05,
    "heading_rad": 2.959135,
    "drift_mean": -0.02226913,
    "drift_var": 8.20059e-05,
    "opacity_var": 0.0036404726,
    "photosphere_crossed": false
  },
  {
    "step": 1340,
    "epoch": 3,
    "batch_in_epoch": 403,
    "theta1": 0.005963,
    "theta2": 0.011537,
    "loss": 0.112611,
    "eta2d_x": 0.486199,
    "eta2d_y": 0.873848,
    "speed": 2.7e-05,
    "heading_rad": 2.138758,
    "drift_mean": 0.01497221,
    "drift_var": 0.0020239281,
    "opacity_var": 0.0023170316,
    "photosphere_crossed": false
  },
  {
    "step": 1344,
    "epoch": 3,
    "batch_in_epoch": 407,
    "theta1": 0.005845,
    "theta2": 0.011577,
    "loss": 0.071593,
    "eta2d_x": 0.465671,
    "eta2d_y": 0.884958,
    "speed": 4.3e-05,
    "heading_rad": 2.581316,
    "drift_mean": -0.01025455,
    "drift_var": 0.0066977385,
    "opacity_var": 0.0041615814,
    "photosphere_crossed": false
  },
  {
    "step": 1348,
    "epoch": 3,
    "batch_in_epoch": 411,
    "theta1": 0.005743,
    "theta2": 0.011673,
    "loss": 0.076461,
    "eta2d_x": 0.443087,
    "eta2d_y": 0.896479,
    "speed": 4.4e-05,
    "heading_rad": 2.517673,
    "drift_mean": 0.00121691,
    "drift_var": 0.0005373646,
    "opacity_var": 0.0038169127,
    "photosphere_crossed": false
  },
  {
    "step": 1352,
    "epoch": 3,
    "batch_in_epoch": 415,
    "theta1": 0.005643,
    "theta2": 0.011772,
    "loss": 0.057158,
    "eta2d_x": 0.420583,
    "eta2d_y": 0.907254,
    "speed": 4.4e-05,
    "heading_rad": 2.097817,
    "drift_mean": -0.0048257,
    "drift_var": 0.0057375912,
    "opacity_var": 0.0039479938,
    "photosphere_crossed": false
  },
  {
    "step": 1356,
    "epoch": 3,
    "batch_in_epoch": 419,
    "theta1": 0.00562,
    "theta2": 0.01184,
    "loss": 0.136079,
    "eta2d_x": 0.41225,
    "eta2d_y": 0.911071,
    "speed": 7e-06,
    "heading_rad": 0.814815,
    "drift_mean": 0.0197303,
    "drift_var": 0.0015374863,
    "opacity_var": 0.0033855032,
    "photosphere_crossed": false
  },
  {
    "step": 1360,
    "epoch": 3,
    "batch_in_epoch": 423,
    "theta1": 0.005624,
    "theta2": 0.011869,
    "loss": 0.055669,
    "eta2d_x": 0.410724,
    "eta2d_y": 0.91176,
    "speed": 1.3e-05,
    "heading_rad": 2.364819,
    "drift_mean": -0.02010264,
    "drift_var": 0.001066211,
    "opacity_var": 0.0016726806,
    "photosphere_crossed": false
  },
  {
    "step": 1364,
    "epoch": 3,
    "batch_in_epoch": 427,
    "theta1": 0.005641,
    "theta2": 0.011898,
    "loss": 0.081609,
    "eta2d_x": 0.41132,
    "eta2d_y": 0.911491,
    "speed": 2.2e-05,
    "heading_rad": 1.454822,
    "drift_mean": 0.00648507,
    "drift_var": 0.0032389506,
    "opacity_var": 0.0023125162,
    "photosphere_crossed": false
  },
  {
    "step": 1368,
    "epoch": 3,
    "batch_in_epoch": 431,
    "theta1": 0.00574,
    "theta2": 0.01189,
    "loss": 0.105407,
    "eta2d_x": 0.426865,
    "eta2d_y": 0.904316,
    "speed": 3.3e-05,
    "heading_rad": -0.959825,
    "drift_mean": 0.00594947,
    "drift_var": 0.0012346276,
    "opacity_var": 0.0023024704,
    "photosphere_crossed": false
  },
  {
    "step": 1372,
    "epoch": 3,
    "batch_in_epoch": 435,
    "theta1": 0.005841,
    "theta2": 0.011781,
    "loss": 0.054825,
    "eta2d_x": 0.449663,
    "eta2d_y": 0.893198,
    "speed": 5e-05,
    "heading_rad": -1.183759,
    "drift_mean": -0.01264555,
    "drift_var": 0.001363591,
    "opacity_var": 0.0013670729,
    "photosphere_crossed": false
  },
  {
    "step": 1376,
    "epoch": 3,
    "batch_in_epoch": 439,
    "theta1": 0.005851,
    "theta2": 0.011603,
    "loss": 0.069895,
    "eta2d_x": 0.464554,
    "eta2d_y": 0.885545,
    "speed": 3.3e-05,
    "heading_rad": -1.201156,
    "drift_mean": 0.00376748,
    "drift_var": 0.0045849993,
    "opacity_var": 0.0028025433,
    "photosphere_crossed": false
  },
  {
    "step": 1380,
    "epoch": 3,
    "batch_in_epoch": 443,
    "theta1": 0.0059,
    "theta2": 0.011615,
    "loss": 0.036891,
    "eta2d_x": 0.470802,
    "eta2d_y": 0.882239,
    "speed": 1e-05,
    "heading_rad": 0.939045,
    "drift_mean": -0.00825083,
    "drift_var": 0.0051494399,
    "opacity_var": 0.0043585374,
    "photosphere_crossed": false
  },
  {
    "step": 1384,
    "epoch": 3,
    "batch_in_epoch": 447,
    "theta1": 0.005758,
    "theta2": 0.011715,
    "loss": 0.063363,
    "eta2d_x": 0.442132,
    "eta2d_y": 0.89695,
    "speed": 5.6e-05,
    "heading_rad": 2.837859,
    "drift_mean": 0.00661797,
    "drift_var": 0.0044275477,
    "opacity_var": 0.0045306834,
    "photosphere_crossed": false
  },
  {
    "step": 1388,
    "epoch": 3,
    "batch_in_epoch": 451,
    "theta1": 0.0055,
    "theta2": 0.011804,
    "loss": 0.108403,
    "eta2d_x": 0.395928,
    "eta2d_y": 0.918281,
    "speed": 5.8e-05,
    "heading_rad": 3.005817,
    "drift_mean": 0.01126007,
    "drift_var": 0.0007611196,
    "opacity_var": 0.0027922081,
    "photosphere_crossed": false
  },
  {
    "step": 1392,
    "epoch": 3,
    "batch_in_epoch": 455,
    "theta1": 0.005416,
    "theta2": 0.011825,
    "loss": 0.061278,
    "eta2d_x": 0.381003,
    "eta2d_y": 0.924574,
    "speed": 2e-06,
    "heading_rad": 1.656281,
    "drift_mean": -0.01178129,
    "drift_var": 0.0058720504,
    "opacity_var": 0.0034422807,
    "photosphere_crossed": false
  },
  {
    "step": 1396,
    "epoch": 3,
    "batch_in_epoch": 459,
    "theta1": 0.005451,
    "theta2": 0.011843,
    "loss": 0.197576,
    "eta2d_x": 0.385531,
    "eta2d_y": 0.922695,
    "speed": 1.9e-05,
    "heading_rad": -0.588039,
    "drift_mean": 0.0340744,
    "drift_var": 0.0057560752,
    "opacity_var": 0.0058162624,
    "photosphere_crossed": false
  },
  {
    "step": 1400,
    "epoch": 3,
    "batch_in_epoch": 463,
    "theta1": 0.005394,
    "theta2": 0.011744,
    "loss": 0.09303,
    "eta2d_x": 0.383033,
    "eta2d_y": 0.923735,
    "speed": 2.3e-05,
    "heading_rad": -2.701876,
    "drift_mean": -0.0261365,
    "drift_var": 0.0053060406,
    "opacity_var": 0.0064148403,
    "photosphere_crossed": false
  },
  {
    "step": 1404,
    "epoch": 3,
    "batch_in_epoch": 467,
    "theta1": 0.005334,
    "theta2": 0.011614,
    "loss": 0.12415,
    "eta2d_x": 0.381891,
    "eta2d_y": 0.924207,
    "speed": 3.7e-05,
    "heading_rad": -2.187093,
    "drift_mean": 0.00778001,
    "drift_var": 0.0071870522,
    "opacity_var": 0.007439493,
    "photosphere_crossed": false
  },
  {
    "step": 1408,
    "epoch": 4,
    "batch_in_epoch": 3,
    "theta1": 0.005181,
    "theta2": 0.011496,
    "loss": 0.09007,
    "eta2d_x": 0.363794,
    "eta2d_y": 0.931479,
    "speed": 4.1e-05,
    "heading_rad": -2.547268,
    "drift_mean": -0.00852002,
    "drift_var": 0.0008587002,
    "opacity_var": 0.0037906153,
    "photosphere_crossed": false
  },
  {
    "step": 1412,
    "epoch": 4,
    "batch_in_epoch": 7,
    "theta1": 0.005022,
    "theta2": 0.011362,
    "loss": 0.033979,
    "eta2d_x": 0.34407,
    "eta2d_y": 0.938944,
    "speed": 5.1e-05,
    "heading_rad": -2.396664,
    "drift_mean": -0.01402269,
    "drift_var": 0.0002340494,
    "opacity_var": 0.0012881742,
    "photosphere_crossed": false
  },
  {
    "step": 1416,
    "epoch": 4,
    "batch_in_epoch": 11,
    "theta1": 0.004962,
    "theta2": 0.011172,
    "loss": 0.041591,
    "eta2d_x": 0.346005,
    "eta2d_y": 0.938233,
    "speed": 3.8e-05,
    "heading_rad": -1.896375,
    "drift_mean": 0.00190287,
    "drift_var": 0.0010375121,
    "opacity_var": 0.0007979624,
    "photosphere_crossed": false
  },
  {
    "step": 1420,
    "epoch": 4,
    "batch_in_epoch": 15,
    "theta1": 0.004916,
    "theta2": 0.011076,
    "loss": 0.112353,
    "eta2d_x": 0.343862,
    "eta2d_y": 0.93902,
    "speed": 2.2e-05,
    "heading_rad": -2.20198,
    "drift_mean": 0.0176905,
    "drift_var": 0.0004276469,
    "opacity_var": 0.0007898459,
    "photosphere_crossed": false
  },
  {
    "step": 1424,
    "epoch": 4,
    "batch_in_epoch": 19,
    "theta1": 0.004879,
    "theta2": 0.010959,
    "loss": 0.048603,
    "eta2d_x": 0.345027,
    "eta2d_y": 0.938593,
    "speed": 4.5e-05,
    "heading_rad": -1.787667,
    "drift_mean": -0.01593737,
    "drift_var": 0.0016177041,
    "opacity_var": 0.0012332957,
    "photosphere_crossed": false
  },
  {
    "step": 1428,
    "epoch": 4,
    "batch_in_epoch": 23,
    "theta1": 0.00479,
    "theta2": 0.010735,
    "loss": 0.045806,
    "eta2d_x": 0.343543,
    "eta2d_y": 0.939137,
    "speed": 5.6e-05,
    "heading_rad": -1.726291,
    "drift_mean": -0.00069919,
    "drift_var": 0.0047977086,
    "opacity_var": 0.0031222055,
    "photosphere_crossed": false
  },
  {
    "step": 1432,
    "epoch": 4,
    "batch_in_epoch": 27,
    "theta1": 0.004764,
    "theta2": 0.010544,
    "loss": 0.077095,
    "eta2d_x": 0.353646,
    "eta2d_y": 0.935379,
    "speed": 2.7e-05,
    "heading_rad": -1.263061,
    "drift_mean": 0.00782222,
    "drift_var": 0.0012991483,
    "opacity_var": 0.0027284029,
    "photosphere_crossed": false
  },
  {
    "step": 1436,
    "epoch": 4,
    "batch_in_epoch": 31,
    "theta1": 0.004907,
    "theta2": 0.010503,
    "loss": 0.0693,
    "eta2d_x": 0.388922,
    "eta2d_y": 0.921271,
    "speed": 4.8e-05,
    "heading_rad": -0.038359,
    "drift_mean": -0.00194881,
    "drift_var": 0.0006636689,
    "opacity_var": 0.0009160453,
    "photosphere_crossed": false
  },
  {
    "step": 1440,
    "epoch": 4,
    "batch_in_epoch": 35,
    "theta1": 0.005126,
    "theta2": 0.010616,
    "loss": 0.038525,
    "eta2d_x": 0.424359,
    "eta2d_y": 0.905494,
    "speed": 5.8e-05,
    "heading_rad": 0.764568,
    "drift_mean": -0.00769385,
    "drift_var": 0.0008881184,
    "opacity_var": 0.0007052286,
    "photosphere_crossed": false
  },
  {
    "step": 1444,
    "epoch": 4,
    "batch_in_epoch": 39,
    "theta1": 0.005244,
    "theta2": 0.010605,
    "loss": 0.040569,
    "eta2d_x": 0.448606,
    "eta2d_y": 0.89373,
    "speed": 2.8e-05,
    "heading_rad": -0.185293,
    "drift_mean": 0.00051114,
    "drift_var": 0.0015495485,
    "opacity_var": 0.0011665062,
    "photosphere_crossed": false
  },
  {
    "step": 1448,
    "epoch": 4,
    "batch_in_epoch": 43,
    "theta1": 0.005242,
    "theta2": 0.010577,
    "loss": 0.153822,
    "eta2d_x": 0.450999,
    "eta2d_y": 0.892524,
    "speed": 3.2e-05,
    "heading_rad": -2.82963,
    "drift_mean": 0.02831335,
    "drift_var": 0.0012533658,
    "opacity_var": 0.0017500908,
    "photosphere_crossed": false
  },
  {
    "step": 1452,
    "epoch": 4,
    "batch_in_epoch": 47,
    "theta1": 0.005163,
    "theta2": 0.010559,
    "loss": 0.097914,
    "eta2d_x": 0.437187,
    "eta2d_y": 0.899371,
    "speed": 1.2e-05,
    "heading_rad": 2.087877,
    "drift_mean": -0.01397723,
    "drift_var": 0.0034060012,
    "opacity_var": 0.002469025,
    "photosphere_crossed": false
  },
  {
    "step": 1456,
    "epoch": 4,
    "batch_in_epoch": 51,
    "theta1": 0.005165,
    "theta2": 0.010487,
    "loss": 0.066587,
    "eta2d_x": 0.444651,
    "eta2d_y": 0.895704,
    "speed": 3.8e-05,
    "heading_rad": -1.320475,
    "drift_mean": -0.00783153,
    "drift_var": 0.000292182,
    "opacity_var": 0.0018810351,
    "photosphere_crossed": false
  },
  {
    "step": 1460,
    "epoch": 4,
    "batch_in_epoch": 55,
    "theta1": 0.005165,
    "theta2": 0.010424,
    "loss": 0.067678,
    "eta2d_x": 0.451007,
    "eta2d_y": 0.89252,
    "speed": 4e-06,
    "heading_rad": 2.765831,
    "drift_mean": 0.0002726,
    "drift_var": 0.0009774598,
    "opacity_var": 0.0006599547,
    "photosphere_crossed": false
  },
  {
    "step": 1464,
    "epoch": 4,
    "batch_in_epoch": 59,
    "theta1": 0.00518,
    "theta2": 0.010426,
    "loss": 0.030799,
    "eta2d_x": 0.453785,
    "eta2d_y": 0.891111,
    "speed": 7e-06,
    "heading_rad": -1.153809,
    "drift_mean": -0.00921964,
    "drift_var": 0.0020818705,
    "opacity_var": 0.0013844768,
    "photosphere_crossed": false
  },
  {
    "step": 1468,
    "epoch": 4,
    "batch_in_epoch": 63,
    "theta1": 0.005165,
    "theta2": 0.010465,
    "loss": 0.039646,
    "eta2d_x": 0.446779,
    "eta2d_y": 0.894644,
    "speed": 2e-05,
    "heading_rad": 2.070063,
    "drift_mean": 0.00221162,
    "drift_var": 0.0002395678,
    "opacity_var": 0.0012787184,
    "photosphere_crossed": false
  },
  {
    "step": 1472,
    "epoch": 4,
    "batch_in_epoch": 67,
    "theta1": 0.005149,
    "theta2": 0.010576,
    "loss": 0.183518,
    "eta2d_x": 0.432741,
    "eta2d_y": 0.901518,
    "speed": 2.7e-05,
    "heading_rad": 2.474052,
    "drift_mean": 0.03596814,
    "drift_var": 0.001630792,
    "opacity_var": 0.0012079124,
    "photosphere_crossed": false
  },
  {
    "step": 1476,
    "epoch": 4,
    "batch_in_epoch": 71,
    "theta1": 0.005079,
    "theta2": 0.010549,
    "loss": 0.026816,
    "eta2d_x": 0.420959,
    "eta2d_y": 0.907079,
    "speed": 1.6e-05,
    "heading_rad": -2.742139,
    "drift_mean": -0.03917564,
    "drift_var": 0.0046533102,
    "opacity_var": 0.0040493534,
    "photosphere_crossed": false
  },
  {
    "step": 1480,
    "epoch": 4,
    "batch_in_epoch": 75,
    "theta1": 0.005063,
    "theta2": 0.010453,
    "loss": 0.044508,
    "eta2d_x": 0.427256,
    "eta2d_y": 0.904131,
    "speed": 3.3e-05,
    "heading_rad": -1.300931,
    "drift_mean": 0.00442313,
    "drift_var": 0.0009603624,
    "opacity_var": 0.0038516024,
    "photosphere_crossed": false
  },
  {
    "step": 1484,
    "epoch": 4,
    "batch_in_epoch": 79,
    "theta1": 0.005004,
    "theta2": 0.010316,
    "loss": 0.052703,
    "eta2d_x": 0.428341,
    "eta2d_y": 0.903617,
    "speed": 4.2e-05,
    "heading_rad": -2.545121,
    "drift_mean": 0.0020487,
    "drift_var": 0.0012601812,
    "opacity_var": 0.0010811788,
    "photosphere_crossed": false
  },
  {
    "step": 1488,
    "epoch": 4,
    "batch_in_epoch": 83,
    "theta1": 0.00488,
    "theta2": 0.010228,
    "loss": 0.026315,
    "eta2d_x": 0.409771,
    "eta2d_y": 0.912189,
    "speed": 3.7e-05,
    "heading_rad": -2.483915,
    "drift_mean": -0.00659706,
    "drift_var": 0.0020057455,
    "opacity_var": 0.0014687727,
    "photosphere_crossed": false
  },
  {
    "step": 1492,
    "epoch": 4,
    "batch_in_epoch": 87,
    "theta1": 0.004835,
    "theta2": 0.010201,
    "loss": 0.014852,
    "eta2d_x": 0.402089,
    "eta2d_y": 0.915601,
    "speed": 2.1e-05,
    "heading_rad": -2.99877,
    "drift_mean": -0.00286566,
    "drift_var": 0.002797913,
    "opacity_var": 0.0024128001,
    "photosphere_crossed": false
  },
  {
    "step": 1496,
    "epoch": 4,
    "batch_in_epoch": 91,
    "theta1": 0.00486,
    "theta2": 0.010154,
    "loss": 0.033684,
    "eta2d_x": 0.412882,
    "eta2d_y": 0.910785,
    "speed": 2.7e-05,
    "heading_rad": -0.520913,
    "drift_mean": 0.0047079,
    "drift_var": 0.0031993163,
    "opacity_var": 0.0027116573,
    "photosphere_crossed": false
  },
  {
    "step": 1500,
    "epoch": 4,
    "batch_in_epoch": 95,
    "theta1": 0.004987,
    "theta2": 0.010182,
    "loss": 0.051139,
    "eta2d_x": 0.438729,
    "eta2d_y": 0.89862,
    "speed": 1.6e-05,
    "heading_rad": 0.6019,
    "drift_mean": 0.0043638,
    "drift_var": 0.0004619144,
    "opacity_var": 0.0023981428,
    "photosphere_crossed": false
  },
  {
    "step": 1504,
    "epoch": 4,
    "batch_in_epoch": 99,
    "theta1": 0.005019,
    "theta2": 0.010199,
    "loss": 0.060156,
    "eta2d_x": 0.444015,
    "eta2d_y": 0.896019,
    "speed": 2.2e-05,
    "heading_rad": 0.757405,
    "drift_mean": 0.00225423,
    "drift_var": 0.0013174276,
    "opacity_var": 0.0008515677,
    "photosphere_crossed": false
  },
  {
    "step": 1508,
    "epoch": 4,
    "batch_in_epoch": 103,
    "theta1": 0.005087,
    "theta2": 0.010146,
    "loss": 0.074094,
    "eta2d_x": 0.464693,
    "eta2d_y": 0.885472,
    "speed": 3.5e-05,
    "heading_rad": -0.930611,
    "drift_mean": 0.00348464,
    "drift_var": 0.0005673081,
    "opacity_var": 0.0008900578,
    "photosphere_crossed": false
  },
  {
    "step": 1512,
    "epoch": 4,
    "batch_in_epoch": 107,
    "theta1": 0.005044,
    "theta2": 0.009915,
    "loss": 0.019589,
    "eta2d_x": 0.482432,
    "eta2d_y": 0.875933,
    "speed": 6.1e-05,
    "heading_rad": -1.699843,
    "drift_mean": -0.01362634,
    "drift_var": 0.0019061719,
    "opacity_var": 0.0012991496,
    "photosphere_crossed": false
  },
  {
    "step": 1516,
    "epoch": 4,
    "batch_in_epoch": 111,
    "theta1": 0.005068,
    "theta2": 0.009791,
    "loss": 0.15226,
    "eta2d_x": 0.503659,
    "eta2d_y": 0.863902,
    "speed": 4.3e-05,
    "heading_rad": -1.010304,
    "drift_mean": 0.03316783,
    "drift_var": 0.0031876297,
    "opacity_var": 0.0027606044,
    "photosphere_crossed": false
  },
  {
    "step": 1520,
    "epoch": 4,
    "batch_in_epoch": 115,
    "theta1": 0.005111,
    "theta2": 0.009682,
    "loss": 0.037625,
    "eta2d_x": 0.528003,
    "eta2d_y": 0.849242,
    "speed": 1.8e-05,
    "heading_rad": -1.558463,
    "drift_mean": -0.02865896,
    "drift_var": 0.0038231978,
    "opacity_var": 0.0039663697,
    "photosphere_crossed": false
  },
  {
    "step": 1524,
    "epoch": 4,
    "batch_in_epoch": 119,
    "theta1": 0.005124,
    "theta2": 0.009651,
    "loss": 0.051959,
    "eta2d_x": 0.535197,
    "eta2d_y": 0.844727,
    "speed": 1.8e-05,
    "heading_rad": 0.651126,
    "drift_mean": 0.0035836,
    "drift_var": 0.0002058915,
    "opacity_var": 0.0039260158,
    "photosphere_crossed": false
  },
  {
    "step": 1528,
    "epoch": 4,
    "batch_in_epoch": 123,
    "theta1": 0.005225,
    "theta2": 0.009736,
    "loss": 0.083817,
    "eta2d_x": 0.544947,
    "eta2d_y": 0.838471,
    "speed": 4e-05,
    "heading_rad": 1.024735,
    "drift_mean": 0.00796462,
    "drift_var": 0.0008647343,
    "opacity_var": 0.0016111645,
    "photosphere_crossed": false
  },
  {
    "step": 1532,
    "epoch": 4,
    "batch_in_epoch": 127,
    "theta1": 0.005175,
    "theta2": 0.009938,
    "loss": 0.050368,
    "eta2d_x": 0.508174,
    "eta2d_y": 0.861255,
    "speed": 4.6e-05,
    "heading_rad": 2.103385,
    "drift_mean": -0.00836227,
    "drift_var": 0.0031636232,
    "opacity_var": 0.001865766,
    "photosphere_crossed": false
  },
  {
    "step": 1536,
    "epoch": 4,
    "batch_in_epoch": 131,
    "theta1": 0.005083,
    "theta2": 0.009944,
    "loss": 0.044827,
    "eta2d_x": 0.487569,
    "eta2d_y": 0.873084,
    "speed": 3.2e-05,
    "heading_rad": -2.535241,
    "drift_mean": -0.00138539,
    "drift_var": 0.0069266086,
    "opacity_var": 0.0046356906,
    "photosphere_crossed": false
  },
  {
    "step": 1540,
    "epoch": 4,
    "batch_in_epoch": 135,
    "theta1": 0.005064,
    "theta2": 0.009853,
    "loss": 0.073249,
    "eta2d_x": 0.494843,
    "eta2d_y": 0.868982,
    "speed": 2.9e-05,
    "heading_rad": -1.753334,
    "drift_mean": 0.00710565,
    "drift_var": 0.0020676417,
    "opacity_var": 0.0041943055,
    "photosphere_crossed": false
  },
  {
    "step": 1544,
    "epoch": 4,
    "batch_in_epoch": 139,
    "theta1": 0.004975,
    "theta2": 0.009835,
    "loss": 0.049409,
    "eta2d_x": 0.476732,
    "eta2d_y": 0.879049,
    "speed": 3.4e-05,
    "heading_rad": 2.641354,
    "drift_mean": -0.00596019,
    "drift_var": 0.0016452942,
    "opacity_var": 0.0017151891,
    "photosphere_crossed": false
  },
  {
    "step": 1548,
    "epoch": 4,
    "batch_in_epoch": 143,
    "theta1": 0.004775,
    "theta2": 0.010044,
    "loss": 0.034159,
    "eta2d_x": 0.404259,
    "eta2d_y": 0.914645,
    "speed": 8e-05,
    "heading_rad": 2.151083,
    "drift_mean": -0.00381239,
    "drift_var": 0.0044534337,
    "opacity_var": 0.0027116468,
    "photosphere_crossed": false
  },
  {
    "step": 1552,
    "epoch": 4,
    "batch_in_epoch": 147,
    "theta1": 0.00457,
    "theta2": 0.010223,
    "loss": 0.039271,
    "eta2d_x": 0.335387,
    "eta2d_y": 0.94208,
    "speed": 5.8e-05,
    "heading_rad": 2.43235,
    "drift_mean": 0.00127801,
    "drift_var": 0.0025302205,
    "opacity_var": 0.0033942657,
    "photosphere_crossed": false
  },
  {
    "step": 1556,
    "epoch": 4,
    "batch_in_epoch": 151,
    "theta1": 0.004405,
    "theta2": 0.010316,
    "loss": 0.023684,
    "eta2d_x": 0.285275,
    "eta2d_y": 0.958446,
    "speed": 2.6e-05,
    "heading_rad": 2.397877,
    "drift_mean": -0.00389677,
    "drift_var": 0.0296755974,
    "opacity_var": 0.0143196477,
    "photosphere_crossed": false
  },
  {
    "step": 1560,
    "epoch": 4,
    "batch_in_epoch": 155,
    "theta1": 0.004402,
    "theta2": 0.010312,
    "loss": 0.098946,
    "eta2d_x": 0.284757,
    "eta2d_y": 0.9586,
    "speed": 4e-06,
    "heading_rad": 0.57015,
    "drift_mean": 0.01881544,
    "drift_var": 0.0016276762,
    "opacity_var": 0.0145423426,
    "photosphere_crossed": false
  },
  {
    "step": 1564,
    "epoch": 4,
    "batch_in_epoch": 159,
    "theta1": 0.004352,
    "theta2": 0.010355,
    "loss": 0.024879,
    "eta2d_x": 0.268599,
    "eta2d_y": 0.963252,
    "speed": 2.4e-05,
    "heading_rad": 2.912497,
    "drift_mean": -0.01851675,
    "drift_var": 0.0018374733,
    "opacity_var": 0.0027149583,
    "photosphere_crossed": false
  },
  {
    "step": 1568,
    "epoch": 4,
    "batch_in_epoch": 163,
    "theta1": 0.004253,
    "theta2": 0.010321,
    "loss": 0.0583,
    "eta2d_x": 0.244698,
    "eta2d_y": 0.969599,
    "speed": 3.8e-05,
    "heading_rad": -2.420715,
    "drift_mean": 0.00835532,
    "drift_var": 0.0132748138,
    "opacity_var": 0.0073614123,
    "photosphere_crossed": false
  },
  {
    "step": 1572,
    "epoch": 4,
    "batch_in_epoch": 167,
    "theta1": 0.004063,
    "theta2": 0.010227,
    "loss": 0.064627,
    "eta2d_x": 0.198391,
    "eta2d_y": 0.980123,
    "speed": 5.2e-05,
    "heading_rad": -2.896358,
    "drift_mean": 0.0015818,
    "drift_var": 0.0004020858,
    "opacity_var": 0.0065153172,
    "photosphere_crossed": false
  },
  {
    "step": 1576,
    "epoch": 4,
    "batch_in_epoch": 171,
    "theta1": 0.003925,
    "theta2": 0.010265,
    "loss": 0.022627,
    "eta2d_x": 0.156779,
    "eta2d_y": 0.987634,
    "speed": 3.7e-05,
    "heading_rad": 2.352468,
    "drift_mean": -0.01049998,
    "drift_var": 0.002696803,
    "opacity_var": 0.0016179675,
    "photosphere_crossed": false
  },
  {
    "step": 1580,
    "epoch": 4,
    "batch_in_epoch": 175,
    "theta1": 0.003858,
    "theta2": 0.010383,
    "loss": 0.058767,
    "eta2d_x": 0.132671,
    "eta2d_y": 0.99116,
    "speed": 4.2e-05,
    "heading_rad": 2.058322,
    "drift_mean": 0.00903484,
    "drift_var": 0.0040443961,
    "opacity_var": 0.0031327267,
    "photosphere_crossed": false
  },
  {
    "step": 1584,
    "epoch": 4,
    "batch_in_epoch": 179,
    "theta1": 0.003776,
    "theta2": 0.010474,
    "loss": 0.0564,
    "eta2d_x": 0.106845,
    "eta2d_y": 0.994276,
    "speed": 3.2e-05,
    "heading_rad": 2.642299,
    "drift_mean": -0.00059162,
    "drift_var": 0.0003272728,
    "opacity_var": 0.0028852874,
    "photosphere_crossed": false
  },
  {
    "step": 1588,
    "epoch": 4,
    "batch_in_epoch": 183,
    "theta1": 0.003715,
    "theta2": 0.010503,
    "loss": 0.048923,
    "eta2d_x": 0.089073,
    "eta2d_y": 0.996025,
    "speed": 1.5e-05,
    "heading_rad": -2.52984,
    "drift_mean": -0.00186934,
    "drift_var": 0.0008124096,
    "opacity_var": 0.0005105727,
    "photosphere_crossed": false
  },
  {
    "step": 1592,
    "epoch": 4,
    "batch_in_epoch": 187,
    "theta1": 0.003748,
    "theta2": 0.010346,
    "loss": 0.067171,
    "eta2d_x": 0.102874,
    "eta2d_y": 0.994694,
    "speed": 4.5e-05,
    "heading_rad": -1.355063,
    "drift_mean": 0.00456198,
    "drift_var": 0.0042513238,
    "opacity_var": 0.0023346262,
    "photosphere_crossed": false
  },
  {
    "step": 1596,
    "epoch": 4,
    "batch_in_epoch": 191,
    "theta1": 0.003838,
    "theta2": 0.010278,
    "loss": 0.040908,
    "eta2d_x": 0.130976,
    "eta2d_y": 0.991386,
    "speed": 2.7e-05,
    "heading_rad": -0.416957,
    "drift_mean": -0.00656561,
    "drift_var": 0.0003802984,
    "opacity_var": 0.0021284539,
    "photosphere_crossed": false
  },
  {
    "step": 1600,
    "epoch": 4,
    "batch_in_epoch": 195,
    "theta1": 0.003864,
    "theta2": 0.010419,
    "loss": 0.069723,
    "eta2d_x": 0.132967,
    "eta2d_y": 0.99112,
    "speed": 3.3e-05,
    "heading_rad": 1.439562,
    "drift_mean": 0.00720361,
    "drift_var": 0.0023835591,
    "opacity_var": 0.001293211,
    "photosphere_crossed": false
  },
  {
    "step": 1604,
    "epoch": 4,
    "batch_in_epoch": 199,
    "theta1": 0.00388,
    "theta2": 0.010501,
    "loss": 0.016639,
    "eta2d_x": 0.134583,
    "eta2d_y": 0.990902,
    "speed": 2.8e-05,
    "heading_rad": 1.013264,
    "drift_mean": -0.01327108,
    "drift_var": 8.13486e-05,
    "opacity_var": 0.0012108311,
    "photosphere_crossed": false
  },
  {
    "step": 1608,
    "epoch": 4,
    "batch_in_epoch": 203,
    "theta1": 0.004008,
    "theta2": 0.010555,
    "loss": 0.058768,
    "eta2d_x": 0.166433,
    "eta2d_y": 0.986053,
    "speed": 4.2e-05,
    "heading_rad": 0.057903,
    "drift_mean": 0.01053244,
    "drift_var": 0.0002469215,
    "opacity_var": 0.0003416614,
    "photosphere_crossed": false
  },
  {
    "step": 1612,
    "epoch": 4,
    "batch_in_epoch": 207,
    "theta1": 0.004119,
    "theta2": 0.010475,
    "loss": 0.080176,
    "eta2d_x": 0.199761,
    "eta2d_y": 0.979845,
    "speed": 3.7e-05,
    "heading_rad": -1.806055,
    "drift_mean": 0.00535191,
    "drift_var": 0.0027234175,
    "opacity_var": 0.0013317829,
    "photosphere_crossed": false
  },
  {
    "step": 1616,
    "epoch": 4,
    "batch_in_epoch": 211,
    "theta1": 0.004043,
    "theta2": 0.010343,
    "loss": 0.064266,
    "eta2d_x": 0.186372,
    "eta2d_y": 0.982479,
    "speed": 2.3e-05,
    "heading_rad": -2.385062,
    "drift_mean": -0.00397755,
    "drift_var": 0.0003433251,
    "opacity_var": 0.00139688,
    "photosphere_crossed": false
  },
  {
    "step": 1620,
    "epoch": 4,
    "batch_in_epoch": 215,
    "theta1": 0.004002,
    "theta2": 0.010228,
    "loss": 0.053708,
    "eta2d_x": 0.18072,
    "eta2d_y": 0.983535,
    "speed": 4.4e-05,
    "heading_rad": -1.768217,
    "drift_mean": -0.0026394,
    "drift_var": 0.000446343,
    "opacity_var": 0.0006043836,
    "photosphere_crossed": false
  },
  {
    "step": 1624,
    "epoch": 4,
    "batch_in_epoch": 219,
    "theta1": 0.003917,
    "theta2": 0.010008,
    "loss": 0.162452,
    "eta2d_x": 0.167016,
    "eta2d_y": 0.985954,
    "speed": 7.5e-05,
    "heading_rad": -1.303107,
    "drift_mean": 0.02718604,
    "drift_var": 0.004089266,
    "opacity_var": 0.0023125666,
    "photosphere_crossed": false
  },
  {
    "step": 1628,
    "epoch": 4,
    "batch_in_epoch": 223,
    "theta1": 0.00401,
    "theta2": 0.009836,
    "loss": 0.123754,
    "eta2d_x": 0.20665,
    "eta2d_y": 0.978415,
    "speed": 2.6e-05,
    "heading_rad": 0.143166,
    "drift_mean": -0.00967458,
    "drift_var": 0.0035509937,
    "opacity_var": 0.0037396411,
    "photosphere_crossed": false
  },
  {
    "step": 1632,
    "epoch": 4,
    "batch_in_epoch": 227,
    "theta1": 0.00415,
    "theta2": 0.009798,
    "loss": 0.02195,
    "eta2d_x": 0.2542,
    "eta2d_y": 0.967152,
    "speed": 3e-05,
    "heading_rad": -0.148501,
    "drift_mean": -0.02545097,
    "drift_var": 0.0012016906,
    "opacity_var": 0.003201292,
    "photosphere_crossed": false
  },
  {
    "step": 1636,
    "epoch": 4,
    "batch_in_epoch": 231,
    "theta1": 0.004241,
    "theta2": 0.009804,
    "loss": 0.061196,
    "eta2d_x": 0.281742,
    "eta2d_y": 0.95949,
    "speed": 2.2e-05,
    "heading_rad": -1.141136,
    "drift_mean": 0.00981134,
    "drift_var": 0.0063060534,
    "opacity_var": 0.0041636286,
    "photosphere_crossed": false
  },
  {
    "step": 1640,
    "epoch": 4,
    "batch_in_epoch": 235,
    "theta1": 0.004254,
    "theta2": 0.009797,
    "loss": 0.03852,
    "eta2d_x": 0.286263,
    "eta2d_y": 0.958151,
    "speed": 1.9e-05,
    "heading_rad": 1.560884,
    "drift_mean": -0.00566887,
    "drift_var": 0.0013918046,
    "opacity_var": 0.0037144007,
    "photosphere_crossed": false
  },
  {
    "step": 1644,
    "epoch": 4,
    "batch_in_epoch": 239,
    "theta1": 0.004344,
    "theta2": 0.009967,
    "loss": 0.108296,
    "eta2d_x": 0.297508,
    "eta2d_y": 0.954719,
    "speed": 6.4e-05,
    "heading_rad": 1.157466,
    "drift_mean": 0.0174439,
    "drift_var": 0.003340604,
    "opacity_var": 0.0023480868,
    "photosphere_crossed": false
  },
  {
    "step": 1648,
    "epoch": 4,
    "batch_in_epoch": 243,
    "theta1": 0.004377,
    "theta2": 0.010132,
    "loss": 0.082818,
    "eta2d_x": 0.292598,
    "eta2d_y": 0.956236,
    "speed": 3.3e-05,
    "heading_rad": 1.518012,
    "drift_mean": -0.00636934,
    "drift_var": 0.0029113548,
    "opacity_var": 0.0031492028,
    "photosphere_crossed": false
  },
  {
    "step": 1652,
    "epoch": 4,
    "batch_in_epoch": 247,
    "theta1": 0.00438,
    "theta2": 0.01026,
    "loss": 0.12383,
    "eta2d_x": 0.283168,
    "eta2d_y": 0.95907,
    "speed": 3e-05,
    "heading_rad": 1.592883,
    "drift_mean": 0.010253,
    "drift_var": 0.0015916937,
    "opacity_var": 0.0027731888,
    "photosphere_crossed": false
  },
  {
    "step": 1656,
    "epoch": 4,
    "batch_in_epoch": 251,
    "theta1": 0.004396,
    "theta2": 0.01041,
    "loss": 0.100731,
    "eta2d_x": 0.275943,
    "eta2d_y": 0.961174,
    "speed": 3.5e-05,
    "heading_rad": 0.615641,
    "drift_mean": -0.00577479,
    "drift_var": 0.0020411014,
    "opacity_var": 0.0018444527,
    "photosphere_crossed": false
  },
  {
    "step": 1660,
    "epoch": 4,
    "batch_in_epoch": 255,
    "theta1": 0.004459,
    "theta2": 0.010487,
    "loss": 0.029304,
    "eta2d_x": 0.286159,
    "eta2d_y": 0.958182,
    "speed": 1.9e-05,
    "heading_rad": 1.400612,
    "drift_mean": -0.01785689,
    "drift_var": 0.0001461178,
    "opacity_var": 0.0010163623,
    "photosphere_crossed": false
  },
  {
    "step": 1664,
    "epoch": 4,
    "batch_in_epoch": 259,
    "theta1": 0.004415,
    "theta2": 0.010622,
    "loss": 0.111566,
    "eta2d_x": 0.265946,
    "eta2d_y": 0.963988,
    "speed": 4.8e-05,
    "heading_rad": 1.685579,
    "drift_mean": 0.02056557,
    "drift_var": 0.0020937045,
    "opacity_var": 0.0014508529,
    "photosphere_crossed": false
  },
  {
    "step": 1668,
    "epoch": 4,
    "batch_in_epoch": 263,
    "theta1": 0.004477,
    "theta2": 0.010881,
    "loss": 0.047364,
    "eta2d_x": 0.263756,
    "eta2d_y": 0.96459,
    "speed": 7.9e-05,
    "heading_rad": 0.923249,
    "drift_mean": -0.01605056,
    "drift_var": 0.0082592563,
    "opacity_var": 0.0049094263,
    "photosphere_crossed": false
  },
  {
    "step": 1672,
    "epoch": 4,
    "batch_in_epoch": 267,
    "theta1": 0.004715,
    "theta2": 0.011071,
    "loss": 0.050619,
    "eta2d_x": 0.303314,
    "eta2d_y": 0.952891,
    "speed": 6e-05,
    "heading_rad": 0.637208,
    "drift_mean": 0.00081385,
    "drift_var": 0.0048158519,
    "opacity_var": 0.0065349471,
    "photosphere_crossed": false
  },
  {
    "step": 1676,
    "epoch": 4,
    "batch_in_epoch": 271,
    "theta1": 0.004769,
    "theta2": 0.010953,
    "loss": 0.090539,
    "eta2d_x": 0.322724,
    "eta2d_y": 0.946493,
    "speed": 7e-05,
    "heading_rad": -1.520429,
    "drift_mean": 0.00998006,
    "drift_var": 0.0019366829,
    "opacity_var": 0.0045018322,
    "photosphere_crossed": false
  },
  {
    "step": 1680,
    "epoch": 4,
    "batch_in_epoch": 275,
    "theta1": 0.004694,
    "theta2": 0.010567,
    "loss": 0.059803,
    "eta2d_x": 0.335674,
    "eta2d_y": 0.941978,
    "speed": 0.000101,
    "heading_rad": -1.728494,
    "drift_mean": -0.00768405,
    "drift_var": 0.0028751993,
    "opacity_var": 0.0022771712,
    "photosphere_crossed": false
  },
  {
    "step": 1684,
    "epoch": 4,
    "batch_in_epoch": 279,
    "theta1": 0.004638,
    "theta2": 0.010204,
    "loss": 0.056046,
    "eta2d_x": 0.354143,
    "eta2d_y": 0.935191,
    "speed": 8.2e-05,
    "heading_rad": -1.802898,
    "drift_mean": -0.0009393,
    "drift_var": 0.000198516,
    "opacity_var": 0.0015927076,
    "photosphere_crossed": false
  },
  {
    "step": 1688,
    "epoch": 4,
    "batch_in_epoch": 283,
    "theta1": 0.004654,
    "theta2": 0.009985,
    "loss": 0.013546,
    "eta2d_x": 0.380211,
    "eta2d_y": 0.9249,
    "speed": 5.2e-05,
    "heading_rad": -1.337921,
    "drift_mean": -0.01062482,
    "drift_var": 0.0032928463,
    "opacity_var": 0.0018397327,
    "photosphere_crossed": false
  },
  {
    "step": 1692,
    "epoch": 4,
    "batch_in_epoch": 287,
    "theta1": 0.00478,
    "theta2": 0.009828,
    "loss": 0.05359,
    "eta2d_x": 0.430279,
    "eta2d_y": 0.902696,
    "speed": 4e-05,
    "heading_rad": -0.737386,
    "drift_mean": 0.01001088,
    "drift_var": 7.2203e-05,
    "opacity_var": 0.0016037332,
    "photosphere_crossed": false
  },
  {
    "step": 1696,
    "epoch": 4,
    "batch_in_epoch": 291,
    "theta1": 0.004823,
    "theta2": 0.009797,
    "loss": 0.082848,
    "eta2d_x": 0.444875,
    "eta2d_y": 0.895592,
    "speed": 1.6e-05,
    "heading_rad": 1.404845,
    "drift_mean": 0.00731459,
    "drift_var": 7.24824e-05,
    "opacity_var": 0.0014410024,
    "photosphere_crossed": false
  },
  {
    "step": 1700,
    "epoch": 4,
    "batch_in_epoch": 295,
    "theta1": 0.004698,
    "theta2": 0.009816,
    "loss": 0.019083,
    "eta2d_x": 0.41046,
    "eta2d_y": 0.911878,
    "speed": 3.3e-05,
    "heading_rad": 2.752371,
    "drift_mean": -0.01594146,
    "drift_var": 0.0021899533,
    "opacity_var": 0.0011349896,
    "photosphere_crossed": false
  },
  {
    "step": 1704,
    "epoch": 4,
    "batch_in_epoch": 299,
    "theta1": 0.004603,
    "theta2": 0.009824,
    "loss": 0.076279,
    "eta2d_x": 0.384312,
    "eta2d_y": 0.923203,
    "speed": 2.3e-05,
    "heading_rad": -2.767556,
    "drift_mean": 0.01429902,
    "drift_var": 0.0003123152,
    "opacity_var": 0.0013195697,
    "photosphere_crossed": false
  },
  {
    "step": 1708,
    "epoch": 4,
    "batch_in_epoch": 303,
    "theta1": 0.004593,
    "theta2": 0.009793,
    "loss": 0.032661,
    "eta2d_x": 0.385143,
    "eta2d_y": 0.922857,
    "speed": 2.1e-05,
    "heading_rad": -0.96942,
    "drift_mean": -0.01090436,
    "drift_var": 0.0004785731,
    "opacity_var": 0.0012802663,
    "photosphere_crossed": false
  },
  {
    "step": 1712,
    "epoch": 4,
    "batch_in_epoch": 307,
    "theta1": 0.004616,
    "theta2": 0.009718,
    "loss": 0.025063,
    "eta2d_x": 0.400267,
    "eta2d_y": 0.916399,
    "speed": 2e-05,
    "heading_rad": -0.685801,
    "drift_mean": -0.00189951,
    "drift_var": 0.0002583312,
    "opacity_var": 0.0003478558,
    "photosphere_crossed": false
  },
  {
    "step": 1716,
    "epoch": 4,
    "batch_in_epoch": 311,
    "theta1": 0.004668,
    "theta2": 0.00961,
    "loss": 0.148309,
    "eta2d_x": 0.427942,
    "eta2d_y": 0.903806,
    "speed": 2.1e-05,
    "heading_rad": -0.697611,
    "drift_mean": 0.03081136,
    "drift_var": 0.002026466,
    "opacity_var": 0.0012876429,
    "photosphere_crossed": false
  },
  {
    "step": 1720,
    "epoch": 4,
    "batch_in_epoch": 315,
    "theta1": 0.004668,
    "theta2": 0.009638,
    "loss": 0.06475,
    "eta2d_x": 0.424472,
    "eta2d_y": 0.905441,
    "speed": 2.5e-05,
    "heading_rad": 0.571855,
    "drift_mean": -0.0208896,
    "drift_var": 0.0030066408,
    "opacity_var": 0.0029205588,
    "photosphere_crossed": false
  },
  {
    "step": 1724,
    "epoch": 4,
    "batch_in_epoch": 319,
    "theta1": 0.004781,
    "theta2": 0.009679,
    "loss": 0.031535,
    "eta2d_x": 0.449127,
    "eta2d_y": 0.893468,
    "speed": 2.7e-05,
    "heading_rad": -0.230256,
    "drift_mean": -0.00830373,
    "drift_var": 0.000885123,
    "opacity_var": 0.002590694,
    "photosphere_crossed": false
  },
  {
    "step": 1728,
    "epoch": 4,
    "batch_in_epoch": 323,
    "theta1": 0.00492,
    "theta2": 0.009756,
    "loss": 0.038279,
    "eta2d_x": 0.473727,
    "eta2d_y": 0.880672,
    "speed": 3.4e-05,
    "heading_rad": 0.367623,
    "drift_mean": 0.00168588,
    "drift_var": 0.0013547105,
    "opacity_var": 0.0012355368,
    "photosphere_crossed": false
  },
  {
    "step": 1732,
    "epoch": 4,
    "batch_in_epoch": 327,
    "theta1": 0.004942,
    "theta2": 0.009732,
    "loss": 0.171355,
    "eta2d_x": 0.482221,
    "eta2d_y": 0.87605,
    "speed": 9e-06,
    "heading_rad": -1.530083,
    "drift_mean": 0.03326895,
    "drift_var": 0.0032471212,
    "opacity_var": 0.0023484488,
    "photosphere_crossed": false
  },
  {
    "step": 1736,
    "epoch": 4,
    "batch_in_epoch": 331,
    "theta1": 0.004912,
    "theta2": 0.009661,
    "loss": 0.055322,
    "eta2d_x": 0.484432,
    "eta2d_y": 0.874829,
    "speed": 7e-06,
    "heading_rad": -3.123591,
    "drift_mean": -0.02900824,
    "drift_var": 0.0021848289,
    "opacity_var": 0.0032839192,
    "photosphere_crossed": false
  },
  {
    "step": 1740,
    "epoch": 4,
    "batch_in_epoch": 335,
    "theta1": 0.004867,
    "theta2": 0.009668,
    "loss": 0.046609,
    "eta2d_x": 0.472341,
    "eta2d_y": 0.881416,
    "speed": 1.3e-05,
    "heading_rad": -1.246144,
    "drift_mean": -0.00217811,
    "drift_var": 0.0005924807,
    "opacity_var": 0.0033150727,
    "photosphere_crossed": false
  },
  {
    "step": 1744,
    "epoch": 4,
    "batch_in_epoch": 339,
    "theta1": 0.004936,
    "theta2": 0.009583,
    "loss": 0.062444,
    "eta2d_x": 0.501216,
    "eta2d_y": 0.865322,
    "speed": 3.8e-05,
    "heading_rad": -0.105273,
    "drift_mean": 0.00395858,
    "drift_var": 0.0014918308,
    "opacity_var": 0.0010298447,
    "photosphere_crossed": false
  },
  {
    "step": 1748,
    "epoch": 4,
    "batch_in_epoch": 343,
    "theta1": 0.005005,
    "theta2": 0.009562,
    "loss": 0.076736,
    "eta2d_x": 0.5205,
    "eta2d_y": 0.853862,
    "speed": 1.2e-05,
    "heading_rad": 0.82317,
    "drift_mean": 0.00357318,
    "drift_var": 0.0006996108,
    "opacity_var": 0.0010007038,
    "photosphere_crossed": false
  },
  {
    "step": 1752,
    "epoch": 4,
    "batch_in_epoch": 347,
    "theta1": 0.005085,
    "theta2": 0.009505,
    "loss": 0.155648,
    "eta2d_x": 0.547278,
    "eta2d_y": 0.836951,
    "speed": 4.8e-05,
    "heading_rad": -1.002396,
    "drift_mean": 0.01972787,
    "drift_var": 0.0018381915,
    "opacity_var": 0.0012496707,
    "photosphere_crossed": false
  },
  {
    "step": 1756,
    "epoch": 4,
    "batch_in_epoch": 351,
    "theta1": 0.005165,
    "theta2": 0.009396,
    "loss": 0.123579,
    "eta2d_x": 0.581819,
    "eta2d_y": 0.813319,
    "speed": 1.4e-05,
    "heading_rad": -1.034288,
    "drift_mean": -0.0080172,
    "drift_var": 0.0050256581,
    "opacity_var": 0.0032271693,
    "photosphere_crossed": false
  },
  {
    "step": 1760,
    "epoch": 4,
    "batch_in_epoch": 355,
    "theta1": 0.005271,
    "theta2": 0.009385,
    "loss": 0.043597,
    "eta2d_x": 0.605799,
    "eta2d_y": 0.795618,
    "speed": 3.1e-05,
    "heading_rad": 0.128645,
    "drift_mean": -0.01999553,
    "drift_var": 0.0044640776,
    "opacity_var": 0.0053317892,
    "photosphere_crossed": false
  },
  {
    "step": 1764,
    "epoch": 4,
    "batch_in_epoch": 359,
    "theta1": 0.005332,
    "theta2": 0.009504,
    "loss": 0.039411,
    "eta2d_x": 0.599937,
    "eta2d_y": 0.800047,
    "speed": 3e-05,
    "heading_rad": 0.787049,
    "drift_mean": -0.00104635,
    "drift_var": 0.0013415077,
    "opacity_var": 0.0032489391,
    "photosphere_crossed": false
  },
  {
    "step": 1768,
    "epoch": 4,
    "batch_in_epoch": 363,
    "theta1": 0.005414,
    "theta2": 0.009568,
    "loss": 0.069211,
    "eta2d_x": 0.606402,
    "eta2d_y": 0.795158,
    "speed": 4.6e-05,
    "heading_rad": 0.786359,
    "drift_mean": 0.00744986,
    "drift_var": 0.000697031,
    "opacity_var": 0.0009265766,
    "photosphere_crossed": false
  },
  {
    "step": 1772,
    "epoch": 4,
    "batch_in_epoch": 367,
    "theta1": 0.005364,
    "theta2": 0.009563,
    "loss": 0.07678,
    "eta2d_x": 0.597528,
    "eta2d_y": 0.801848,
    "speed": 1.4e-05,
    "heading_rad": -2.807016,
    "drift_mean": 0.00189224,
    "drift_var": 0.0028895209,
    "opacity_var": 0.0016124895,
    "photosphere_crossed": false
  },
  {
    "step": 1776,
    "epoch": 4,
    "batch_in_epoch": 371,
    "theta1": 0.005332,
    "theta2": 0.00953,
    "loss": 0.039878,
    "eta2d_x": 0.596195,
    "eta2d_y": 0.802839,
    "speed": 2e-05,
    "heading_rad": -1.745804,
    "drift_mean": -0.00922534,
    "drift_var": 0.0005302044,
    "opacity_var": 0.0015785703,
    "photosphere_crossed": false
  },
  {
    "step": 1780,
    "epoch": 4,
    "batch_in_epoch": 375,
    "theta1": 0.005253,
    "theta2": 0.009497,
    "loss": 0.042264,
    "eta2d_x": 0.584979,
    "eta2d_y": 0.811048,
    "speed": 2.2e-05,
    "heading_rad": -2.835787,
    "drift_mean": 0.00059648,
    "drift_var": 0.0002606672,
    "opacity_var": 0.000404924,
    "photosphere_crossed": false
  },
  {
    "step": 1784,
    "epoch": 4,
    "batch_in_epoch": 379,
    "theta1": 0.005162,
    "theta2": 0.009421,
    "loss": 0.049981,
    "eta2d_x": 0.577297,
    "eta2d_y": 0.816535,
    "speed": 2.5e-05,
    "heading_rad": -2.767288,
    "drift_mean": 0.00192911,
    "drift_var": 0.00861336,
    "opacity_var": 0.0040855653,
    "photosphere_crossed": false
  },
  {
    "step": 1788,
    "epoch": 4,
    "batch_in_epoch": 383,
    "theta1": 0.004951,
    "theta2": 0.009318,
    "loss": 0.086281,
    "eta2d_x": 0.544425,
    "eta2d_y": 0.83881,
    "speed": 8.1e-05,
    "heading_rad": -2.378959,
    "drift_mean": 0.00907492,
    "drift_var": 0.0005640552,
    "opacity_var": 0.0041205568,
    "photosphere_crossed": false
  },
  {
    "step": 1792,
    "epoch": 4,
    "batch_in_epoch": 387,
    "theta1": 0.004745,
    "theta2": 0.00922,
    "loss": 0.096652,
    "eta2d_x": 0.506316,
    "eta2d_y": 0.862348,
    "speed": 2.1e-05,
    "heading_rad": -2.668628,
    "drift_mean": 0.00259298,
    "drift_var": 0.0029941197,
    "opacity_var": 0.0015963593,
    "photosphere_crossed": false
  },
  {
    "step": 1796,
    "epoch": 4,
    "batch_in_epoch": 391,
    "theta1": 0.004689,
    "theta2": 0.009216,
    "loss": 0.051902,
    "eta2d_x": 0.491215,
    "eta2d_y": 0.871038,
    "speed": 1.2e-05,
    "heading_rad": -1.866989,
    "drift_mean": -0.01118759,
    "drift_var": 0.0015325847,
    "opacity_var": 0.0021080224,
    "photosphere_crossed": false
  },
  {
    "step": 1800,
    "epoch": 4,
    "batch_in_epoch": 395,
    "theta1": 0.004804,
    "theta2": 0.009125,
    "loss": 0.035852,
    "eta2d_x": 0.538785,
    "eta2d_y": 0.842443,
    "speed": 3.7e-05,
    "heading_rad": -0.703106,
    "drift_mean": -0.00401254,
    "drift_var": 0.0011363798,
    "opacity_var": 0.0013728714,
    "photosphere_crossed": false
  },
  {
    "step": 1804,
    "epoch": 4,
    "batch_in_epoch": 399,
    "theta1": 0.004876,
    "theta2": 0.009125,
    "loss": 0.043138,
    "eta2d_x": 0.557736,
    "eta2d_y": 0.830018,
    "speed": 2.6e-05,
    "heading_rad": 0.551285,
    "drift_mean": 0.00182149,
    "drift_var": 0.0008730218,
    "opacity_var": 0.0009152146,
    "photosphere_crossed": false
  },
  {
    "step": 1808,
    "epoch": 4,
    "batch_in_epoch": 403,
    "theta1": 0.004932,
    "theta2": 0.009239,
    "loss": 0.040371,
    "eta2d_x": 0.552646,
    "eta2d_y": 0.833416,
    "speed": 2.2e-05,
    "heading_rad": 0.533419,
    "drift_mean": -0.00069184,
    "drift_var": 0.0026928428,
    "opacity_var": 0.0015863707,
    "photosphere_crossed": false
  },
  {
    "step": 1812,
    "epoch": 4,
    "batch_in_epoch": 407,
    "theta1": 0.004899,
    "theta2": 0.009142,
    "loss": 0.113786,
    "eta2d_x": 0.560686,
    "eta2d_y": 0.828029,
    "speed": 5.8e-05,
    "heading_rad": -2.471171,
    "drift_mean": 0.01835376,
    "drift_var": 8.46322e-05,
    "opacity_var": 0.001374673,
    "photosphere_crossed": false
  },
  {
    "step": 1816,
    "epoch": 4,
    "batch_in_epoch": 411,
    "theta1": 0.004823,
    "theta2": 0.00901,
    "loss": 0.055928,
    "eta2d_x": 0.564358,
    "eta2d_y": 0.82553,
    "speed": 3.3e-05,
    "heading_rad": -2.853372,
    "drift_mean": -0.01446432,
    "drift_var": 0.0032636873,
    "opacity_var": 0.0017782001,
    "photosphere_crossed": false
  },
  {
    "step": 1820,
    "epoch": 4,
    "batch_in_epoch": 415,
    "theta1": 0.004751,
    "theta2": 0.00897,
    "loss": 0.036871,
    "eta2d_x": 0.552017,
    "eta2d_y": 0.833833,
    "speed": 8e-06,
    "heading_rad": 2.436103,
    "drift_mean": -0.00476436,
    "drift_var": 0.0029353771,
    "opacity_var": 0.0028066791,
    "photosphere_crossed": false
  },
  {
    "step": 1824,
    "epoch": 4,
    "batch_in_epoch": 419,
    "theta1": 0.004752,
    "theta2": 0.008975,
    "loss": 0.061231,
    "eta2d_x": 0.551325,
    "eta2d_y": 0.834291,
    "speed": 4e-06,
    "heading_rad": -1.089387,
    "drift_mean": 0.0060901,
    "drift_var": 0.0001170321,
    "opacity_var": 0.0017826539,
    "photosphere_crossed": false
  },
  {
    "step": 1828,
    "epoch": 4,
    "batch_in_epoch": 423,
    "theta1": 0.004736,
    "theta2": 0.009025,
    "loss": 0.079633,
    "eta2d_x": 0.537551,
    "eta2d_y": 0.843231,
    "speed": 1.2e-05,
    "heading_rad": 1.659124,
    "drift_mean": 0.00460042,
    "drift_var": 0.0052968707,
    "opacity_var": 0.0030607869,
    "photosphere_crossed": false
  },
  {
    "step": 1832,
    "epoch": 4,
    "batch_in_epoch": 427,
    "theta1": 0.004811,
    "theta2": 0.009147,
    "loss": 0.101913,
    "eta2d_x": 0.536642,
    "eta2d_y": 0.84381,
    "speed": 4.3e-05,
    "heading_rad": 0.802027,
    "drift_mean": 0.00557,
    "drift_var": 0.0011104243,
    "opacity_var": 0.0028586461,
    "photosphere_crossed": false
  },
  {
    "step": 1836,
    "epoch": 4,
    "batch_in_epoch": 431,
    "theta1": 0.004966,
    "theta2": 0.009145,
    "loss": 0.062857,
    "eta2d_x": 0.576832,
    "eta2d_y": 0.816863,
    "speed": 2.7e-05,
    "heading_rad": -0.391714,
    "drift_mean": -0.00976392,
    "drift_var": 0.0066629986,
    "opacity_var": 0.0041670634,
    "photosphere_crossed": false
  },
  {
    "step": 1840,
    "epoch": 4,
    "batch_in_epoch": 435,
    "theta1": 0.004924,
    "theta2": 0.009125,
    "loss": 0.091849,
    "eta2d_x": 0.569989,
    "eta2d_y": 0.821652,
    "speed": 3.9e-05,
    "heading_rad": -2.321013,
    "drift_mean": 0.00724792,
    "drift_var": 0.0014921976,
    "opacity_var": 0.0038163601,
    "photosphere_crossed": false
  },
  {
    "step": 1844,
    "epoch": 4,
    "batch_in_epoch": 439,
    "theta1": 0.004789,
    "theta2": 0.009018,
    "loss": 0.082437,
    "eta2d_x": 0.553489,
    "eta2d_y": 0.832857,
    "speed": 7.1e-05,
    "heading_rad": -2.73006,
    "drift_mean": -0.00235309,
    "drift_var": 0.000414283,
    "opacity_var": 0.0015426695,
    "photosphere_crossed": false
  },
  {
    "step": 1848,
    "epoch": 4,
    "batch_in_epoch": 443,
    "theta1": 0.004536,
    "theta2": 0.009044,
    "loss": 0.04715,
    "eta2d_x": 0.473572,
    "eta2d_y": 0.880755,
    "speed": 5e-05,
    "heading_rad": 2.6944,
    "drift_mean": -0.00882156,
    "drift_var": 0.0002868403,
    "opacity_var": 0.000574859,
    "photosphere_crossed": false
  },
  {
    "step": 1852,
    "epoch": 4,
    "batch_in_epoch": 447,
    "theta1": 0.004443,
    "theta2": 0.009141,
    "loss": 0.126717,
    "eta2d_x": 0.427122,
    "eta2d_y": 0.904194,
    "speed": 2.7e-05,
    "heading_rad": 2.567633,
    "drift_mean": 0.01989162,
    "drift_var": 0.0019893427,
    "opacity_var": 0.0012030882,
    "photosphere_crossed": false
  },
  {
    "step": 1856,
    "epoch": 4,
    "batch_in_epoch": 451,
    "theta1": 0.004538,
    "theta2": 0.009288,
    "loss": 0.079933,
    "eta2d_x": 0.434914,
    "eta2d_y": 0.900472,
    "speed": 5e-05,
    "heading_rad": 0.570873,
    "drift_mean": -0.01169607,
    "drift_var": 0.0006873676,
    "opacity_var": 0.0015524144,
    "photosphere_crossed": false
  },
  {
    "step": 1860,
    "epoch": 4,
    "batch_in_epoch": 455,
    "theta1": 0.004667,
    "theta2": 0.009294,
    "loss": 0.045313,
    "eta2d_x": 0.472609,
    "eta2d_y": 0.881272,
    "speed": 2.5e-05,
    "heading_rad": -0.113235,
    "drift_mean": -0.00865486,
    "drift_var": 0.0006754849,
    "opacity_var": 0.001706809,
    "photosphere_crossed": false
  },
  {
    "step": 1864,
    "epoch": 4,
    "batch_in_epoch": 459,
    "theta1": 0.004633,
    "theta2": 0.009206,
    "loss": 0.047028,
    "eta2d_x": 0.476456,
    "eta2d_y": 0.879198,
    "speed": 3.4e-05,
    "heading_rad": -2.42306,
    "drift_mean": 0.00042858,
    "drift_var": 0.0039096183,
    "opacity_var": 0.0021123334,
    "photosphere_crossed": false
  },
  {
    "step": 1868,
    "epoch": 4,
    "batch_in_epoch": 463,
    "theta1": 0.004446,
    "theta2": 0.009154,
    "loss": 0.073826,
    "eta2d_x": 0.426034,
    "eta2d_y": 0.904707,
    "speed": 3.6e-05,
    "heading_rad": -3.00525,
    "drift_mean": 0.00669964,
    "drift_var": 0.0005636266,
    "opacity_var": 0.0020259014,
    "photosphere_crossed": false
  },
  {
    "step": 1872,
    "epoch": 4,
    "batch_in_epoch": 467,
    "theta1": 0.00439,
    "theta2": 0.009155,
    "loss": 0.162738,
    "eta2d_x": 0.407038,
    "eta2d_y": 0.913411,
    "speed": 2.3e-05,
    "heading_rad": -0.040947,
    "drift_mean": 0.02222794,
    "drift_var": 0.0021515632,
    "opacity_var": 0.0020316281,
    "photosphere_crossed": false
  },
  {
    "step": 1876,
    "epoch": 5,
    "batch_in_epoch": 3,
    "theta1": 0.004372,
    "theta2": 0.009078,
    "loss": 0.036107,
    "eta2d_x": 0.412734,
    "eta2d_y": 0.910852,
    "speed": 9e-06,
    "heading_rad": -0.961304,
    "drift_mean": -0.03165782,
    "drift_var": 0.0007483716,
    "opacity_var": 0.0019408477,
    "photosphere_crossed": false
  },
  {
    "step": 1880,
    "epoch": 5,
    "batch_in_epoch": 7,
    "theta1": 0.004447,
    "theta2": 0.009162,
    "loss": 0.033854,
    "eta2d_x": 0.42507,
    "eta2d_y": 0.90516,
    "speed": 4.7e-05,
    "heading_rad": 1.016066,
    "drift_mean": -0.00056308,
    "drift_var": 0.0002519742,
    "opacity_var": 0.0016162414,
    "photosphere_crossed": false
  },
  {
    "step": 1884,
    "epoch": 5,
    "batch_in_epoch": 11,
    "theta1": 0.004588,
    "theta2": 0.009346,
    "loss": 0.021129,
    "eta2d_x": 0.441596,
    "eta2d_y": 0.897214,
    "speed": 4.6e-05,
    "heading_rad": 0.947753,
    "drift_mean": -0.00318137,
    "drift_var": 0.0017454723,
    "opacity_var": 0.001082625,
    "photosphere_crossed": false
  },
  {
    "step": 1888,
    "epoch": 5,
    "batch_in_epoch": 15,
    "theta1": 0.004704,
    "theta2": 0.009392,
    "loss": 0.027239,
    "eta2d_x": 0.468154,
    "eta2d_y": 0.883647,
    "speed": 2.7e-05,
    "heading_rad": 0.049882,
    "drift_mean": 0.00152761,
    "drift_var": 8.50964e-05,
    "opacity_var": 0.0008206391,
    "photosphere_crossed": false
  },
  {
    "step": 1892,
    "epoch": 5,
    "batch_in_epoch": 19,
    "theta1": 0.004801,
    "theta2": 0.009295,
    "loss": 0.061435,
    "eta2d_x": 0.50961,
    "eta2d_y": 0.860405,
    "speed": 5.1e-05,
    "heading_rad": -1.167085,
    "drift_mean": 0.00854887,
    "drift_var": 0.0001068446,
    "opacity_var": 0.000243143,
    "photosphere_crossed": true
  },
  {
    "step": 1896,
    "epoch": 5,
    "batch_in_epoch": 23,
    "theta1": 0.004814,
    "theta2": 0.009094,
    "loss": 0.035121,
    "eta2d_x": 0.546806,
    "eta2d_y": 0.837259,
    "speed": 5.4e-05,
    "heading_rad": -1.497657,
    "drift_mean": -0.00657844,
    "drift_var": 9.63996e-05,
    "opacity_var": 0.0001476298,
    "photosphere_crossed": true
  },
  {
    "step": 1900,
    "epoch": 5,
    "batch_in_epoch": 27,
    "theta1": 0.004861,
    "theta2": 0.008896,
    "loss": 0.052054,
    "eta2d_x": 0.596202,
    "eta2d_y": 0.802834,
    "speed": 3.9e-05,
    "heading_rad": -1.193773,
    "drift_mean": 0.00423315,
    "drift_var": 0.0003226522,
    "opacity_var": 0.0002122391,
    "photosphere_crossed": true
  },
  {
    "step": 1904,
    "epoch": 5,
    "batch_in_epoch": 31,
    "theta1": 0.004905,
    "theta2": 0.008783,
    "loss": 0.03161,
    "eta2d_x": 0.62997,
    "eta2d_y": 0.776619,
    "speed": 3e-05,
    "heading_rad": -1.236767,
    "drift_mean": -0.00511082,
    "drift_var": 0.0005061248,
    "opacity_var": 0.0003965125,
    "photosphere_crossed": false
  },
  {
    "step": 1908,
    "epoch": 5,
    "batch_in_epoch": 35,
    "theta1": 0.004932,
    "theta2": 0.008729,
    "loss": 0.018357,
    "eta2d_x": 0.647773,
    "eta2d_y": 0.761834,
    "speed": 1.2e-05,
    "heading_rad": 1.133483,
    "drift_mean": -0.00331331,
    "drift_var": 0.0006327691,
    "opacity_var": 0.0006605008,
    "photosphere_crossed": false
  },
  {
    "step": 1912,
    "epoch": 5,
    "batch_in_epoch": 39,
    "theta1": 0.004885,
    "theta2": 0.008833,
    "loss": 0.044555,
    "eta2d_x": 0.614581,
    "eta2d_y": 0.788853,
    "speed": 4.9e-05,
    "heading_rad": 2.66431,
    "drift_mean": 0.00654954,
    "drift_var": 0.0002878551,
    "opacity_var": 0.000441981,
    "photosphere_crossed": false
  },
  {
    "step": 1916,
    "epoch": 5,
    "batch_in_epoch": 43,
    "theta1": 0.004806,
    "theta2": 0.008898,
    "loss": 0.036344,
    "eta2d_x": 0.581056,
    "eta2d_y": 0.813864,
    "speed": 2.5e-05,
    "heading_rad": -2.961423,
    "drift_mean": -0.00205287,
    "drift_var": 0.0001319878,
    "opacity_var": 0.0002195779,
    "photosphere_crossed": true
  },
  {
    "step": 1920,
    "epoch": 5,
    "batch_in_epoch": 47,
    "theta1": 0.004787,
    "theta2": 0.008859,
    "loss": 0.040826,
    "eta2d_x": 0.583325,
    "eta2d_y": 0.812239,
    "speed": 1.4e-05,
    "heading_rad": -1.929042,
    "drift_mean": 0.00112066,
    "drift_var": 0.0005213366,
    "opacity_var": 0.0002978688,
    "photosphere_crossed": true
  },
  {
    "step": 1924,
    "epoch": 5,
    "batch_in_epoch": 51,
    "theta1": 0.004805,
    "theta2": 0.008829,
    "loss": 0.040782,
    "eta2d_x": 0.594281,
    "eta2d_y": 0.804257,
    "speed": 1.2e-05,
    "heading_rad": 1.417903,
    "drift_mean": -1.102e-05,
    "drift_var": 0.0009532163,
    "opacity_var": 0.0006588576,
    "photosphere_crossed": false
  },
  {
    "step": 1928,
    "epoch": 5,
    "batch_in_epoch": 55,
    "theta1": 0.004732,
    "theta2": 0.008903,
    "loss": 0.093288,
    "eta2d_x": 0.55925,
    "eta2d_y": 0.828999,
    "speed": 4.3e-05,
    "heading_rad": 2.326497,
    "drift_mean": 0.01312641,
    "drift_var": 0.0005399426,
    "opacity_var": 0.0007363702,
    "photosphere_crossed": false
  },
  {
    "step": 1932,
    "epoch": 5,
    "batch_in_epoch": 59,
    "theta1": 0.00471,
    "theta2": 0.008957,
    "loss": 0.021487,
    "eta2d_x": 0.542676,
    "eta2d_y": 0.839942,
    "speed": 6e-06,
    "heading_rad": 2.140477,
    "drift_mean": -0.01795027,
    "drift_var": 0.001419249,
    "opacity_var": 0.0011664608,
    "photosphere_crossed": false
  },
  {
    "step": 1936,
    "epoch": 5,
    "batch_in_epoch": 63,
    "theta1": 0.004748,
    "theta2": 0.008917,
    "loss": 0.010171,
    "eta2d_x": 0.561097,
    "eta2d_y": 0.82775,
    "speed": 3.3e-05,
    "heading_rad": -0.579478,
    "drift_mean": -0.00282903,
    "drift_var": 0.0003855877,
    "opacity_var": 0.0012171743,
    "photosphere_crossed": false
  },
  {
    "step": 1940,
    "epoch": 5,
    "batch_in_epoch": 67,
    "theta1": 0.004865,
    "theta2": 0.008928,
    "loss": 0.038024,
    "eta2d_x": 0.590925,
    "eta2d_y": 0.806726,
    "speed": 4.4e-05,
    "heading_rad": 0.666147,
    "drift_mean": 0.00696328,
    "drift_var": 0.0007967692,
    "opacity_var": 0.0006335941,
    "photosphere_crossed": false
  },
  {
    "step": 1944,
    "epoch": 5,
    "batch_in_epoch": 71,
    "theta1": 0.005005,
    "theta2": 0.009028,
    "loss": 0.087224,
    "eta2d_x": 0.607267,
    "eta2d_y": 0.794498,
    "speed": 1.8e-05,
    "heading_rad": 0.547562,
    "drift_mean": 0.01230007,
    "drift_var": 0.0014561225,
    "opacity_var": 0.0011777,
    "photosphere_crossed": false
  },
  {
    "step": 1948,
    "epoch": 5,
    "batch_in_epoch": 75,
    "theta1": 0.005078,
    "theta2": 0.009057,
    "loss": 0.072106,
    "eta2d_x": 0.618858,
    "eta2d_y": 0.785503,
    "speed": 1.4e-05,
    "heading_rad": -0.438356,
    "drift_mean": -0.00377956,
    "drift_var": 0.0017868915,
    "opacity_var": 0.0015068855,
    "photosphere_crossed": false
  },
  {
    "step": 1952,
    "epoch": 5,
    "batch_in_epoch": 79,
    "theta1": 0.005083,
    "theta2": 0.008971,
    "loss": 0.069549,
    "eta2d_x": 0.635635,
    "eta2d_y": 0.77199,
    "speed": 4.4e-05,
    "heading_rad": -1.900247,
    "drift_mean": -0.00063933,
    "drift_var": 0.0005801999,
    "opacity_var": 0.0014655969,
    "photosphere_crossed": false
  },
  {
    "step": 1956,
    "epoch": 5,
    "batch_in_epoch": 83,
    "theta1": 0.004943,
    "theta2": 0.008837,
    "loss": 0.032364,
    "eta2d_x": 0.628446,
    "eta2d_y": 0.777853,
    "speed": 5.2e-05,
    "heading_rad": -2.602533,
    "drift_mean": -0.00929604,
    "drift_var": 0.0011205194,
    "opacity_var": 0.0010148102,
    "photosphere_crossed": false
  },
  {
    "step": 1960,
    "epoch": 5,
    "batch_in_epoch": 87,
    "theta1": 0.004726,
    "theta2": 0.008893,
    "loss": 0.07321,
    "eta2d_x": 0.559389,
    "eta2d_y": 0.828906,
    "speed": 4.6e-05,
    "heading_rad": 2.390313,
    "drift_mean": 0.01021141,
    "drift_var": 0.0018058035,
    "opacity_var": 0.0014270909,
    "photosphere_crossed": false
  },
  {
    "step": 1964,
    "epoch": 5,
    "batch_in_epoch": 91,
    "theta1": 0.004672,
    "theta2": 0.008882,
    "loss": 0.064624,
    "eta2d_x": 0.545832,
    "eta2d_y": 0.837895,
    "speed": 3e-05,
    "heading_rad": -2.258308,
    "drift_mean": -0.00214646,
    "drift_var": 0.000656961,
    "opacity_var": 0.0014122436,
    "photosphere_crossed": false
  },
  {
    "step": 1968,
    "epoch": 5,
    "batch_in_epoch": 95,
    "theta1": 0.004612,
    "theta2": 0.008847,
    "loss": 0.052387,
    "eta2d_x": 0.534143,
    "eta2d_y": 0.845394,
    "speed": 1.1e-05,
    "heading_rad": 2.461181,
    "drift_mean": -0.00305925,
    "drift_var": 0.0006148179,
    "opacity_var": 0.0005676568,
    "photosphere_crossed": false
  },
  {
    "step": 1972,
    "epoch": 5,
    "batch_in_epoch": 99,
    "theta1": 0.0046,
    "theta2": 0.00888,
    "loss": 0.021574,
    "eta2d_x": 0.524147,
    "eta2d_y": 0.851628,
    "speed": 8e-06,
    "heading_rad": 1.740692,
    "drift_mean": -0.0077033,
    "drift_var": 0.0004014892,
    "opacity_var": 0.0006736732,
    "photosphere_crossed": false
  },
  {
    "step": 1976,
    "epoch": 5,
    "batch_in_epoch": 103,
    "theta1": 0.004543,
    "theta2": 0.008927,
    "loss": 0.033132,
    "eta2d_x": 0.496606,
    "eta2d_y": 0.867976,
    "speed": 1.8e-05,
    "heading_rad": 2.133599,
    "drift_mean": 0.00288948,
    "drift_var": 0.0017672526,
    "opacity_var": 0.000996812,
    "photosphere_crossed": false
  },
  {
    "step": 1980,
    "epoch": 5,
    "batch_in_epoch": 107,
    "theta1": 0.004615,
    "theta2": 0.008988,
    "loss": 0.058021,
    "eta2d_x": 0.508506,
    "eta2d_y": 0.861058,
    "speed": 2.6e-05,
    "heading_rad": 0.33968,
    "drift_mean": 0.00622216,
    "drift_var": 0.0006823028,
    "opacity_var": 0.0010939841,
    "photosphere_crossed": false
  },
  {
    "step": 1984,
    "epoch": 5,
    "batch_in_epoch": 111,
    "theta1": 0.004651,
    "theta2": 0.009052,
    "loss": 0.089557,
    "eta2d_x": 0.507872,
    "eta2d_y": 0.861432,
    "speed": 3.3e-05,
    "heading_rad": 1.919915,
    "drift_mean": 0.00788413,
    "drift_var": 0.0009360577,
    "opacity_var": 0.0007261607,
    "photosphere_crossed": false
  },
  {
    "step": 1988,
    "epoch": 5,
    "batch_in_epoch": 115,
    "theta1": 0.004528,
    "theta2": 0.009073,
    "loss": 0.051017,
    "eta2d_x": 0.465806,
    "eta2d_y": 0.884887,
    "speed": 2.9e-05,
    "heading_rad": -3.134617,
    "drift_mean": -0.00963501,
    "drift_var": 0.0014363183,
    "opacity_var": 0.0012649968,
    "photosphere_crossed": false
  },
  {
    "step": 1992,
    "epoch": 5,
    "batch_in_epoch": 119,
    "theta1": 0.004466,
    "theta2": 0.00893,
    "loss": 0.124098,
    "eta2d_x": 0.470423,
    "eta2d_y": 0.882441,
    "speed": 4.7e-05,
    "heading_rad": -1.648073,
    "drift_mean": 0.01827028,
    "drift_var": 0.0023770221,
    "opacity_var": 0.0020217053,
    "photosphere_crossed": false
  },
  {
    "step": 1996,
    "epoch": 5,
    "batch_in_epoch": 123,
    "theta1": 0.004439,
    "theta2": 0.008892,
    "loss": 0.019352,
    "eta2d_x": 0.468043,
    "eta2d_y": 0.883706,
    "speed": 1.1e-05,
    "heading_rad": 1.950283,
    "drift_mean": -0.02618647,
    "drift_var": 0.000858493,
    "opacity_var": 0.0019909219,
    "photosphere_crossed": false
  },
  {
    "step": 2000,
    "epoch": 5,
    "batch_in_epoch": 127,
    "theta1": 0.004489,
    "theta2": 0.009023,
    "loss": 0.087991,
    "eta2d_x": 0.461841,
    "eta2d_y": 0.886963,
    "speed": 3.7e-05,
    "heading_rad": 0.848444,
    "drift_mean": 0.01715969,
    "drift_var": 0.007514859,
    "opacity_var": 0.0050748939,
    "photosphere_crossed": false
  },
  {
    "step": 2004,
    "epoch": 5,
    "batch_in_epoch": 131,
    "theta1": 0.004508,
    "theta2": 0.009018,
    "loss": 0.046747,
    "eta2d_x": 0.468973,
    "eta2d_y": 0.883213,
    "speed": 1.6e-05,
    "heading_rad": -1.170047,
    "drift_mean": -0.01031097,
    "drift_var": 0.0012123843,
    "opacity_var": 0.0041564807,
    "photosphere_crossed": false
  },
  {
    "step": 2008,
    "epoch": 5,
    "batch_in_epoch": 135,
    "theta1": 0.004367,
    "theta2": 0.008986,
    "loss": 0.034017,
    "eta2d_x": 0.425933,
    "eta2d_y": 0.904755,
    "speed": 5.7e-05,
    "heading_rad": -3.115184,
    "drift_mean": -0.00318249,
    "drift_var": 0.0003515115,
    "opacity_var": 0.0012385918,
    "photosphere_crossed": false
  },
  {
    "step": 2012,
    "epoch": 5,
    "batch_in_epoch": 139,
    "theta1": 0.004107,
    "theta2": 0.00904,
    "loss": 0.023647,
    "eta2d_x": 0.318664,
    "eta2d_y": 0.947868,
    "speed": 6.9e-05,
    "heading_rad": 2.91306,
    "drift_mean": -0.00259245,
    "drift_var": 0.0020099607,
    "opacity_var": 0.001059489,
    "photosphere_crossed": false
  },
  {
    "step": 2016,
    "epoch": 5,
    "batch_in_epoch": 143,
    "theta1": 0.003828,
    "theta2": 0.009065,
    "loss": 0.043776,
    "eta2d_x": 0.198335,
    "eta2d_y": 0.980134,
    "speed": 4.3e-05,
    "heading_rad": -3.141219,
    "drift_mean": 0.00503219,
    "drift_var": 0.0035521546,
    "opacity_var": 0.0026172966,
    "photosphere_crossed": false
  },
  {
    "step": 2020,
    "epoch": 5,
    "batch_in_epoch": 147,
    "theta1": 0.003707,
    "theta2": 0.009031,
    "loss": 0.029429,
    "eta2d_x": 0.146775,
    "eta2d_y": 0.98917,
    "speed": 1.2e-05,
    "heading_rad": -2.413669,
    "drift_mean": -0.00358682,
    "drift_var": 0.0016339848,
    "opacity_var": 0.0024156642,
    "photosphere_crossed": false
  },
  {
    "step": 2024,
    "epoch": 5,
    "batch_in_epoch": 151,
    "theta1": 0.003662,
    "theta2": 0.008958,
    "loss": 0.085599,
    "eta2d_x": 0.130531,
    "eta2d_y": 0.991444,
    "speed": 1.7e-05,
    "heading_rad": -2.388566,
    "drift_mean": 0.01404262,
    "drift_var": 0.002755847,
    "opacity_var": 0.0020215612,
    "photosphere_crossed": false
  },
  {
    "step": 2028,
    "epoch": 5,
    "batch_in_epoch": 155,
    "theta1": 0.003596,
    "theta2": 0.008879,
    "loss": 0.034647,
    "eta2d_x": 0.102539,
    "eta2d_y": 0.994729,
    "speed": 2.9e-05,
    "heading_rad": -2.239912,
    "drift_mean": -0.01273803,
    "drift_var": 0.0004433592,
    "opacity_var": 0.0015946738,
    "photosphere_crossed": false
  },
  {
    "step": 2032,
    "epoch": 5,
    "batch_in_epoch": 159,
    "theta1": 0.003519,
    "theta2": 0.008696,
    "loss": 0.068595,
    "eta2d_x": 0.070208,
    "eta2d_y": 0.997532,
    "speed": 6.6e-05,
    "heading_rad": -1.954081,
    "drift_mean": 0.00848697,
    "drift_var": 0.0003643189,
    "opacity_var": 0.0008525358,
    "photosphere_crossed": false
  },
  {
    "step": 2036,
    "epoch": 5,
    "batch_in_epoch": 163,
    "theta1": 0.003518,
    "theta2": 0.008495,
    "loss": 0.09639,
    "eta2d_x": 0.078636,
    "eta2d_y": 0.996903,
    "speed": 2.8e-05,
    "heading_rad": -1.377832,
    "drift_mean": 0.00694871,
    "drift_var": 0.0012973956,
    "opacity_var": 0.0007391122,
    "photosphere_crossed": false
  },
  {
    "step": 2040,
    "epoch": 5,
    "batch_in_epoch": 167,
    "theta1": 0.003489,
    "theta2": 0.008338,
    "loss": 0.024663,
    "eta2d_x": 0.066629,
    "eta2d_y": 0.997778,
    "speed": 5e-05,
    "heading_rad": -1.8128,
    "drift_mean": -0.01793174,
    "drift_var": 0.0023357603,
    "opacity_var": 0.0019220428,
    "photosphere_crossed": false
  },
  {
    "step": 2044,
    "epoch": 5,
    "batch_in_epoch": 171,
    "theta1": 0.003545,
    "theta2": 0.008227,
    "loss": 0.052819,
    "eta2d_x": 0.114614,
    "eta2d_y": 0.99341,
    "speed": 2.1e-05,
    "heading_rad": -0.111756,
    "drift_mean": 0.00703902,
    "drift_var": 8.0019e-06,
    "opacity_var": 0.0011810855,
    "photosphere_crossed": false
  },
  {
    "step": 2048,
    "epoch": 5,
    "batch_in_epoch": 175,
    "theta1": 0.00351,
    "theta2": 0.008231,
    "loss": 0.031315,
    "eta2d_x": 0.088322,
    "eta2d_y": 0.996092,
    "speed": 2e-05,
    "heading_rad": -2.982419,
    "drift_mean": -0.00537609,
    "drift_var": 0.0029100485,
    "opacity_var": 0.0018199311,
    "photosphere_crossed": false
  },
  {
    "step": 2052,
    "epoch": 5,
    "batch_in_epoch": 179,
    "theta1": 0.003513,
    "theta2": 0.008285,
    "loss": 0.042949,
    "eta2d_x": 0.087138,
    "eta2d_y": 0.996196,
    "speed": 2.8e-05,
    "heading_rad": 1.116758,
    "drift_mean": 0.00290849,
    "drift_var": 0.0015328257,
    "opacity_var": 0.0019912606,
    "photosphere_crossed": false
  },
  {
    "step": 2056,
    "epoch": 5,
    "batch_in_epoch": 183,
    "theta1": 0.003603,
    "theta2": 0.008325,
    "loss": 0.041385,
    "eta2d_x": 0.146844,
    "eta2d_y": 0.98916,
    "speed": 2.3e-05,
    "heading_rad": -0.381921,
    "drift_mean": -0.00039092,
    "drift_var": 1.73711e-05,
    "opacity_var": 0.0007082085,
    "photosphere_crossed": false
  },
  {
    "step": 2060,
    "epoch": 5,
    "batch_in_epoch": 187,
    "theta1": 0.003809,
    "theta2": 0.008336,
    "loss": 0.062424,
    "eta2d_x": 0.28114,
    "eta2d_y": 0.959667,
    "speed": 5.8e-05,
    "heading_rad": 0.353535,
    "drift_mean": 0.00525972,
    "drift_var": 0.000408778,
    "opacity_var": 0.0002117382,
    "photosphere_crossed": true
  },
  {
    "step": 2064,
    "epoch": 5,
    "batch_in_epoch": 191,
    "theta1": 0.004053,
    "theta2": 0.008411,
    "loss": 0.092949,
    "eta2d_x": 0.404366,
    "eta2d_y": 0.914597,
    "speed": 7.1e-05,
    "heading_rad": 0.146136,
    "drift_mean": 0.0076313,
    "drift_var": 0.0017925141,
    "opacity_var": 0.0009801334,
    "photosphere_crossed": false
  },
  {
    "step": 2068,
    "epoch": 5,
    "batch_in_epoch": 195,
    "theta1": 0.004175,
    "theta2": 0.008357,
    "loss": 0.039286,
    "eta2d_x": 0.477524,
    "eta2d_y": 0.878619,
    "speed": 3e-05,
    "heading_rad": -0.971983,
    "drift_mean": -0.0134158,
    "drift_var": 0.0016112779,
    "opacity_var": 0.0016745712,
    "photosphere_crossed": false
  },
  {
    "step": 2072,
    "epoch": 5,
    "batch_in_epoch": 199,
    "theta1": 0.00411,
    "theta2": 0.008195,
    "loss": 0.01822,
    "eta2d_x": 0.489536,
    "eta2d_y": 0.871983,
    "speed": 5e-05,
    "heading_rad": -2.24264,
    "drift_mean": -0.00526641,
    "drift_var": 0.0023406475,
    "opacity_var": 0.0018422321,
    "photosphere_crossed": false
  },
  {
    "step": 2076,
    "epoch": 5,
    "batch_in_epoch": 203,
    "theta1": 0.004068,
    "theta2": 0.008132,
    "loss": 0.032431,
    "eta2d_x": 0.486281,
    "eta2d_y": 0.873803,
    "speed": 1.9e-05,
    "heading_rad": 1.055646,
    "drift_mean": 0.00355262,
    "drift_var": 0.0001124545,
    "opacity_var": 0.0011189803,
    "photosphere_crossed": false
  },
  {
    "step": 2080,
    "epoch": 5,
    "batch_in_epoch": 207,
    "theta1": 0.004211,
    "theta2": 0.008315,
    "loss": 0.057264,
    "eta2d_x": 0.5054,
    "eta2d_y": 0.862885,
    "speed": 6.7e-05,
    "heading_rad": 0.727435,
    "drift_mean": 0.00620838,
    "drift_var": 0.0005843685,
    "opacity_var": 0.0008784259,
    "photosphere_crossed": false
  },
  {
    "step": 2084,
    "epoch": 5,
    "batch_in_epoch": 211,
    "theta1": 0.004404,
    "theta2": 0.008447,
    "loss": 0.061175,
    "eta2d_x": 0.551652,
    "eta2d_y": 0.834075,
    "speed": 5.1e-05,
    "heading_rad": 0.750249,
    "drift_mean": 0.00097765,
    "drift_var": 0.0005405938,
    "opacity_var": 0.0005064564,
    "photosphere_crossed": false
  },
  {
    "step": 2088,
    "epoch": 5,
    "batch_in_epoch": 215,
    "theta1": 0.004429,
    "theta2": 0.008544,
    "loss": 0.151237,
    "eta2d_x": 0.537334,
    "eta2d_y": 0.843369,
    "speed": 1.3e-05,
    "heading_rad": 1.55237,
    "drift_mean": 0.02251546,
    "drift_var": 0.0062294382,
    "opacity_var": 0.0031130722,
    "photosphere_crossed": false
  },
  {
    "step": 2092,
    "epoch": 5,
    "batch_in_epoch": 219,
    "theta1": 0.004488,
    "theta2": 0.008548,
    "loss": 0.070662,
    "eta2d_x": 0.55769,
    "eta2d_y": 0.83005,
    "speed": 4.1e-05,
    "heading_rad": -0.588155,
    "drift_mean": -0.02014376,
    "drift_var": 0.0041304359,
    "opacity_var": 0.0050392159,
    "photosphere_crossed": false
  },
  {
    "step": 2096,
    "epoch": 5,
    "batch_in_epoch": 223,
    "theta1": 0.0047,
    "theta2": 0.008521,
    "loss": 0.024816,
    "eta2d_x": 0.632102,
    "eta2d_y": 0.774885,
    "speed": 5.2e-05,
    "heading_rad": 0.018124,
    "drift_mean": -0.01146141,
    "drift_var": 0.000152542,
    "opacity_var": 0.0035055946,
    "photosphere_crossed": false
  },
  {
    "step": 2100,
    "epoch": 5,
    "batch_in_epoch": 227,
    "theta1": 0.004857,
    "theta2": 0.008592,
    "loss": 0.012625,
    "eta2d_x": 0.658398,
    "eta2d_y": 0.75267,
    "speed": 3.4e-05,
    "heading_rad": 0.716383,
    "drift_mean": -0.00304788,
    "drift_var": 0.0004113369,
    "opacity_var": 0.0002959093,
    "photosphere_crossed": true
  },
  {
    "step": 2104,
    "epoch": 5,
    "batch_in_epoch": 231,
    "theta1": 0.005059,
    "theta2": 0.008675,
    "loss": 0.028408,
    "eta2d_x": 0.688218,
    "eta2d_y": 0.725504,
    "speed": 5e-05,
    "heading_rad": 0.280297,
    "drift_mean": 0.0039458,
    "drift_var": 0.0006781051,
    "opacity_var": 0.0005251575,
    "photosphere_crossed": false
  },
  {
    "step": 2108,
    "epoch": 5,
    "batch_in_epoch": 235,
    "theta1": 0.005122,
    "theta2": 0.008661,
    "loss": 0.017132,
    "eta2d_x": 0.704538,
    "eta2d_y": 0.709667,
    "speed": 1e-05,
    "heading_rad": -1.212162,
    "drift_mean": -0.002819,
    "drift_var": 0.0020858863,
    "opacity_var": 0.0013591667,
    "photosphere_crossed": false
  },
  {
    "step": 2112,
    "epoch": 5,
    "batch_in_epoch": 239,
    "theta1": 0.005164,
    "theta2": 0.008622,
    "loss": 0.044537,
    "eta2d_x": 0.720863,
    "eta2d_y": 0.693078,
    "speed": 3.3e-05,
    "heading_rad": -0.699983,
    "drift_mean": 0.00685138,
    "drift_var": 0.0096802455,
    "opacity_var": 0.0053248821,
    "photosphere_crossed": false
  },
  {
    "step": 2116,
    "epoch": 5,
    "batch_in_epoch": 243,
    "theta1": 0.005201,
    "theta2": 0.008554,
    "loss": 0.032295,
    "eta2d_x": 0.741543,
    "eta2d_y": 0.670905,
    "speed": 1.4e-05,
    "heading_rad": -1.943224,
    "drift_mean": -0.0030605,
    "drift_var": 0.0005612952,
    "opacity_var": 0.0049589723,
    "photosphere_crossed": false
  },
  {
    "step": 2120,
    "epoch": 5,
    "batch_in_epoch": 247,
    "theta1": 0.005125,
    "theta2": 0.008402,
    "loss": 0.03745,
    "eta2d_x": 0.759616,
    "eta2d_y": 0.650372,
    "speed": 5.1e-05,
    "heading_rad": -2.031509,
    "drift_mean": 0.00128868,
    "drift_var": 0.0009359864,
    "opacity_var": 0.0006727573,
    "photosphere_crossed": false
  },
  {
    "step": 2124,
    "epoch": 5,
    "batch_in_epoch": 251,
    "theta1": 0.005056,
    "theta2": 0.008286,
    "loss": 0.083955,
    "eta2d_x": 0.772608,
    "eta2d_y": 0.634884,
    "speed": 2.4e-05,
    "heading_rad": -3.11535,
    "drift_mean": 0.01162628,
    "drift_var": 0.0030679881,
    "opacity_var": 0.0019292503,
    "photosphere_crossed": false
  },
  {
    "step": 2128,
    "epoch": 5,
    "batch_in_epoch": 255,
    "theta1": 0.004899,
    "theta2": 0.008311,
    "loss": 0.0509,
    "eta2d_x": 0.734519,
    "eta2d_y": 0.678588,
    "speed": 5.2e-05,
    "heading_rad": 2.823255,
    "drift_mean": -0.00826373,
    "drift_var": 0.0001556236,
    "opacity_var": 0.0015228549,
    "photosphere_crossed": false
  },
  {
    "step": 2132,
    "epoch": 5,
    "batch_in_epoch": 259,
    "theta1": 0.004832,
    "theta2": 0.008426,
    "loss": 0.020704,
    "eta2d_x": 0.690531,
    "eta2d_y": 0.723303,
    "speed": 3.1e-05,
    "heading_rad": 1.684228,
    "drift_mean": -0.00754901,
    "drift_var": 0.0002434577,
    "opacity_var": 0.0007155563,
    "photosphere_crossed": false
  },
  {
    "step": 2136,
    "epoch": 5,
    "batch_in_epoch": 263,
    "theta1": 0.004809,
    "theta2": 0.008477,
    "loss": 0.086477,
    "eta2d_x": 0.672477,
    "eta2d_y": 0.740118,
    "speed": 2.5e-05,
    "heading_rad": -0.960695,
    "drift_mean": 0.01644321,
    "drift_var": 0.0003646005,
    "opacity_var": 0.000437507,
    "photosphere_crossed": false
  },
  {
    "step": 2140,
    "epoch": 5,
    "batch_in_epoch": 267,
    "theta1": 0.004854,
    "theta2": 0.008376,
    "loss": 0.034657,
    "eta2d_x": 0.707852,
    "eta2d_y": 0.70636,
    "speed": 3.2e-05,
    "heading_rad": -1.369608,
    "drift_mean": -0.01295497,
    "drift_var": 0.0017408327,
    "opacity_var": 0.0011307411,
    "photosphere_crossed": false
  },
  {
    "step": 2144,
    "epoch": 5,
    "batch_in_epoch": 271,
    "theta1": 0.004803,
    "theta2": 0.008322,
    "loss": 0.04796,
    "eta2d_x": 0.708796,
    "eta2d_y": 0.705414,
    "speed": 2.7e-05,
    "heading_rad": -2.79467,
    "drift_mean": 0.00332561,
    "drift_var": 0.0006588471,
    "opacity_var": 0.0013913503,
    "photosphere_crossed": false
  },
  {
    "step": 2148,
    "epoch": 5,
    "batch_in_epoch": 275,
    "theta1": 0.004679,
    "theta2": 0.008392,
    "loss": 0.061501,
    "eta2d_x": 0.657652,
    "eta2d_y": 0.753322,
    "speed": 6.1e-05,
    "heading_rad": 2.349411,
    "drift_mean": 0.00338538,
    "drift_var": 0.0007594341,
    "opacity_var": 0.0009117974,
    "photosphere_crossed": false
  },
  {
    "step": 2152,
    "epoch": 5,
    "batch_in_epoch": 279,
    "theta1": 0.004542,
    "theta2": 0.008476,
    "loss": 0.02495,
    "eta2d_x": 0.593771,
    "eta2d_y": 0.804634,
    "speed": 3.6e-05,
    "heading_rad": 2.641639,
    "drift_mean": -0.00913788,
    "drift_var": 0.000543042,
    "opacity_var": 0.0006751806,
    "photosphere_crossed": false
  },
  {
    "step": 2156,
    "epoch": 5,
    "batch_in_epoch": 283,
    "theta1": 0.004463,
    "theta2": 0.008564,
    "loss": 0.034203,
    "eta2d_x": 0.545222,
    "eta2d_y": 0.838292,
    "speed": 2.3e-05,
    "heading_rad": 1.887978,
    "drift_mean": 0.00231346,
    "drift_var": 0.0012512629,
    "opacity_var": 0.0009902238,
    "photosphere_crossed": false
  },
  {
    "step": 2160,
    "epoch": 5,
    "batch_in_epoch": 287,
    "theta1": 0.004509,
    "theta2": 0.008625,
    "loss": 0.031059,
    "eta2d_x": 0.547078,
    "eta2d_y": 0.837082,
    "speed": 3.5e-05,
    "heading_rad": 0.75892,
    "drift_mean": -0.00078609,
    "drift_var": 0.0001498071,
    "opacity_var": 0.0006531137,
    "photosphere_crossed": false
  },
  {
    "step": 2164,
    "epoch": 5,
    "batch_in_epoch": 291,
    "theta1": 0.004567,
    "theta2": 0.008688,
    "loss": 0.024598,
    "eta2d_x": 0.552629,
    "eta2d_y": 0.833427,
    "speed": 2.1e-05,
    "heading_rad": 0.437642,
    "drift_mean": -0.00161523,
    "drift_var": 0.0003795239,
    "opacity_var": 0.0002446389,
    "photosphere_crossed": true
  },
  {
    "step": 2168,
    "epoch": 5,
    "batch_in_epoch": 295,
    "theta1": 0.004679,
    "theta2": 0.008714,
    "loss": 0.059843,
    "eta2d_x": 0.582112,
    "eta2d_y": 0.813109,
    "speed": 2.5e-05,
    "heading_rad": 0.2461,
    "drift_mean": 0.00881115,
    "drift_var": 0.0004968835,
    "opacity_var": 0.0004337581,
    "photosphere_crossed": false
  },
  {
    "step": 2172,
    "epoch": 5,
    "batch_in_epoch": 299,
    "theta1": 0.004743,
    "theta2": 0.008759,
    "loss": 0.068384,
    "eta2d_x": 0.591432,
    "eta2d_y": 0.806355,
    "speed": 3e-05,
    "heading_rad": 0.79993,
    "drift_mean": 0.00213526,
    "drift_var": 0.0009507369,
    "opacity_var": 0.000727844,
    "photosphere_crossed": false
  },
  {
    "step": 2176,
    "epoch": 5,
    "batch_in_epoch": 303,
    "theta1": 0.004805,
    "theta2": 0.008844,
    "loss": 0.067134,
    "eta2d_x": 0.591246,
    "eta2d_y": 0.806491,
    "speed": 1.9e-05,
    "heading_rad": 1.665112,
    "drift_mean": -0.00031255,
    "drift_var": 0.0004869064,
    "opacity_var": 0.0008212046,
    "photosphere_crossed": false
  },
  {
    "step": 2180,
    "epoch": 5,
    "batch_in_epoch": 307,
    "theta1": 0.004815,
    "theta2": 0.008852,
    "loss": 0.017854,
    "eta2d_x": 0.59241,
    "eta2d_y": 0.805637,
    "speed": 2.2e-05,
    "heading_rad": -1.127297,
    "drift_mean": -0.01231979,
    "drift_var": 0.0017314089,
    "opacity_var": 0.0013635676,
    "photosphere_crossed": false
  },
  {
    "step": 2184,
    "epoch": 5,
    "batch_in_epoch": 311,
    "theta1": 0.004842,
    "theta2": 0.008896,
    "loss": 0.077478,
    "eta2d_x": 0.591003,
    "eta2d_y": 0.806669,
    "speed": 3.3e-05,
    "heading_rad": 0.707946,
    "drift_mean": 0.01490596,
    "drift_var": 0.0003034961,
    "opacity_var": 0.0010691372,
    "photosphere_crossed": false
  },
  {
    "step": 2188,
    "epoch": 5,
    "batch_in_epoch": 315,
    "theta1": 0.004947,
    "theta2": 0.008917,
    "loss": 0.086616,
    "eta2d_x": 0.613925,
    "eta2d_y": 0.789364,
    "speed": 3.4e-05,
    "heading_rad": 0.580345,
    "drift_mean": 0.00228436,
    "drift_var": 0.0014722786,
    "opacity_var": 0.0014045692,
    "photosphere_crossed": false
  },
  {
    "step": 2192,
    "epoch": 5,
    "batch_in_epoch": 319,
    "theta1": 0.004946,
    "theta2": 0.008977,
    "loss": 0.049206,
    "eta2d_x": 0.602214,
    "eta2d_y": 0.798334,
    "speed": 3.9e-05,
    "heading_rad": 1.998629,
    "drift_mean": -0.00935237,
    "drift_var": 0.0014185945,
    "opacity_var": 0.0015209647,
    "photosphere_crossed": false
  },
  {
    "step": 2196,
    "epoch": 5,
    "batch_in_epoch": 323,
    "theta1": 0.0049,
    "theta2": 0.009179,
    "loss": 0.038288,
    "eta2d_x": 0.554551,
    "eta2d_y": 0.83215,
    "speed": 5.4e-05,
    "heading_rad": 1.667808,
    "drift_mean": -0.00272959,
    "drift_var": 0.0002088917,
    "opacity_var": 0.0011430094,
    "photosphere_crossed": false
  },
  {
    "step": 2200,
    "epoch": 5,
    "batch_in_epoch": 327,
    "theta1": 0.004865,
    "theta2": 0.009361,
    "loss": 0.054839,
    "eta2d_x": 0.516119,
    "eta2d_y": 0.856517,
    "speed": 6.4e-05,
    "heading_rad": 1.939986,
    "drift_mean": 0.00413771,
    "drift_var": 0.0001612768,
    "opacity_var": 0.0002102352,
    "photosphere_crossed": true
  },
  {
    "step": 2204,
    "epoch": 5,
    "batch_in_epoch": 331,
    "theta1": 0.004807,
    "theta2": 0.009705,
    "loss": 0.100666,
    "eta2d_x": 0.452336,
    "eta2d_y": 0.891847,
    "speed": 8.2e-05,
    "heading_rad": 1.772542,
    "drift_mean": 0.01145689,
    "drift_var": 0.0013310237,
    "opacity_var": 0.0006755309,
    "photosphere_crossed": false
  },
  {
    "step": 2208,
    "epoch": 5,
    "batch_in_epoch": 335,
    "theta1": 0.004697,
    "theta2": 0.009905,
    "loss": 0.044159,
    "eta2d_x": 0.399956,
    "eta2d_y": 0.916534,
    "speed": 4.1e-05,
    "heading_rad": 2.344522,
    "drift_mean": -0.01412672,
    "drift_var": 0.0035468976,
    "opacity_var": 0.0023248667,
    "photosphere_crossed": false
  },
  {
    "step": 2212,
    "epoch": 5,
    "batch_in_epoch": 339,
    "theta1": 0.004586,
    "theta2": 0.009957,
    "loss": 0.059742,
    "eta2d_x": 0.365285,
    "eta2d_y": 0.930896,
    "speed": 3e-05,
    "heading_rad": 3.132315,
    "drift_mean": 0.00389573,
    "drift_var": 5.73164e-05,
    "opacity_var": 0.0020687658,
    "photosphere_crossed": false
  },
  {
    "step": 2216,
    "epoch": 5,
    "batch_in_epoch": 343,
    "theta1": 0.004553,
    "theta2": 0.010088,
    "loss": 0.03651,
    "eta2d_x": 0.343534,
    "eta2d_y": 0.93914,
    "speed": 5.1e-05,
    "heading_rad": 1.405154,
    "drift_mean": -0.00580803,
    "drift_var": 0.0009166774,
    "opacity_var": 0.0005037754,
    "photosphere_crossed": false
  },
  {
    "step": 2220,
    "epoch": 5,
    "batch_in_epoch": 347,
    "theta1": 0.004651,
    "theta2": 0.010329,
    "loss": 0.034012,
    "eta2d_x": 0.346048,
    "eta2d_y": 0.938217,
    "speed": 5.6e-05,
    "heading_rad": 1.225727,
    "drift_mean": -0.0006246,
    "drift_var": 0.0092638261,
    "opacity_var": 0.0045680596,
    "photosphere_crossed": false
  },
  {
    "step": 2224,
    "epoch": 5,
    "batch_in_epoch": 351,
    "theta1": 0.004596,
    "theta2": 0.010478,
    "loss": 0.08308,
    "eta2d_x": 0.320068,
    "eta2d_y": 0.947395,
    "speed": 2.4e-05,
    "heading_rad": -3.126645,
    "drift_mean": 0.01226716,
    "drift_var": 0.0002340905,
    "opacity_var": 0.0043768165,
    "photosphere_crossed": false
  },
  {
    "step": 2228,
    "epoch": 5,
    "batch_in_epoch": 355,
    "theta1": 0.004508,
    "theta2": 0.01046,
    "loss": 0.029498,
    "eta2d_x": 0.300156,
    "eta2d_y": 0.95389,
    "speed": 1.5e-05,
    "heading_rad": -2.645863,
    "drift_mean": -0.01339551,
    "drift_var": 0.0017577244,
    "opacity_var": 0.0027328886,
    "photosphere_crossed": false
  },
  {
    "step": 2232,
    "epoch": 5,
    "batch_in_epoch": 359,
    "theta1": 0.004582,
    "theta2": 0.010402,
    "loss": 0.057241,
    "eta2d_x": 0.322887,
    "eta2d_y": 0.946437,
    "speed": 4.7e-05,
    "heading_rad": -1.178532,
    "drift_mean": 0.0069356,
    "drift_var": 0.0028686501,
    "opacity_var": 0.0022572176,
    "photosphere_crossed": false
  },
  {
    "step": 2236,
    "epoch": 5,
    "batch_in_epoch": 363,
    "theta1": 0.004676,
    "theta2": 0.010326,
    "loss": 0.034666,
    "eta2d_x": 0.352149,
    "eta2d_y": 0.935944,
    "speed": 2.5e-05,
    "heading_rad": -0.174892,
    "drift_mean": -0.0056437,
    "drift_var": 0.0021517831,
    "opacity_var": 0.002446843,
    "photosphere_crossed": false
  },
  {
    "step": 2240,
    "epoch": 5,
    "batch_in_epoch": 367,
    "theta1": 0.004743,
    "theta2": 0.010344,
    "loss": 0.00629,
    "eta2d_x": 0.366552,
    "eta2d_y": 0.930398,
    "speed": 1.5e-05,
    "heading_rad": 0.601277,
    "drift_mean": -0.00709398,
    "drift_var": 0.0003856517,
    "opacity_var": 0.0012502406,
    "photosphere_crossed": false
  },
  {
    "step": 2244,
    "epoch": 5,
    "batch_in_epoch": 371,
    "theta1": 0.00477,
    "theta2": 0.010337,
    "loss": 0.036361,
    "eta2d_x": 0.373586,
    "eta2d_y": 0.927596,
    "speed": 2e-05,
    "heading_rad": -1.992856,
    "drift_mean": 0.00751769,
    "drift_var": 0.0012582228,
    "opacity_var": 0.0007886923,
    "photosphere_crossed": false
  },
  {
    "step": 2248,
    "epoch": 5,
    "batch_in_epoch": 375,
    "theta1": 0.004636,
    "theta2": 0.010261,
    "loss": 0.014386,
    "eta2d_x": 0.348531,
    "eta2d_y": 0.937297,
    "speed": 4.2e-05,
    "heading_rad": -2.404838,
    "drift_mean": -0.00549372,
    "drift_var": 0.0006803863,
    "opacity_var": 0.0009146424,
    "photosphere_crossed": false
  },
  {
    "step": 2252,
    "epoch": 5,
    "batch_in_epoch": 379,
    "theta1": 0.00458,
    "theta2": 0.010084,
    "loss": 0.057534,
    "eta2d_x": 0.350963,
    "eta2d_y": 0.936389,
    "speed": 6.5e-05,
    "heading_rad": -1.805133,
    "drift_mean": 0.01078711,
    "drift_var": 0.0004716994,
    "opacity_var": 0.0008063564,
    "photosphere_crossed": false
  },
  {
    "step": 2256,
    "epoch": 5,
    "batch_in_epoch": 383,
    "theta1": 0.004418,
    "theta2": 0.009886,
    "loss": 0.048891,
    "eta2d_x": 0.326211,
    "eta2d_y": 0.945297,
    "speed": 5.9e-05,
    "heading_rad": -2.02563,
    "drift_mean": -0.00216089,
    "drift_var": 0.0018359889,
    "opacity_var": 0.001154712,
    "photosphere_crossed": false
  },
  {
    "step": 2260,
    "epoch": 5,
    "batch_in_epoch": 387,
    "theta1": 0.004382,
    "theta2": 0.009741,
    "loss": 0.032454,
    "eta2d_x": 0.330402,
    "eta2d_y": 0.94384,
    "speed": 2.7e-05,
    "heading_rad": -1.591292,
    "drift_mean": -0.00410925,
    "drift_var": 0.0004646745,
    "opacity_var": 0.0010540232,
    "photosphere_crossed": false
  },
  {
    "step": 2264,
    "epoch": 5,
    "batch_in_epoch": 391,
    "theta1": 0.004395,
    "theta2": 0.009715,
    "loss": 0.053097,
    "eta2d_x": 0.337077,
    "eta2d_y": 0.941477,
    "speed": 3e-06,
    "heading_rad": 0.182197,
    "drift_mean": 0.00516075,
    "drift_var": 0.0031828846,
    "opacity_var": 0.0016839448,
    "photosphere_crossed": false
  },
  {
    "step": 2268,
    "epoch": 5,
    "batch_in_epoch": 395,
    "theta1": 0.004407,
    "theta2": 0.009698,
    "loss": 0.040616,
    "eta2d_x": 0.342578,
    "eta2d_y": 0.93949,
    "speed": 2.8e-05,
    "heading_rad": -1.486789,
    "drift_mean": -0.00312019,
    "drift_var": 0.0006763606,
    "opacity_var": 0.0017919402,
    "photosphere_crossed": false
  },
  {
    "step": 2272,
    "epoch": 5,
    "batch_in_epoch": 399,
    "theta1": 0.004403,
    "theta2": 0.009575,
    "loss": 0.025555,
    "eta2d_x": 0.35513,
    "eta2d_y": 0.934817,
    "speed": 3.1e-05,
    "heading_rad": -1.457098,
    "drift_mean": -0.00376522,
    "drift_var": 0.0014839236,
    "opacity_var": 0.0011861169,
    "photosphere_crossed": false
  },
  {
    "step": 2276,
    "epoch": 5,
    "batch_in_epoch": 403,
    "theta1": 0.004401,
    "theta2": 0.009492,
    "loss": 0.029418,
    "eta2d_x": 0.364549,
    "eta2d_y": 0.931184,
    "speed": 2.3e-05,
    "heading_rad": -1.507288,
    "drift_mean": 0.00096573,
    "drift_var": 0.0017353503,
    "opacity_var": 0.0014663106,
    "photosphere_crossed": false
  },
  {
    "step": 2280,
    "epoch": 5,
    "batch_in_epoch": 407,
    "theta1": 0.004376,
    "theta2": 0.009331,
    "loss": 0.022507,
    "eta2d_x": 0.377095,
    "eta2d_y": 0.926175,
    "speed": 5.7e-05,
    "heading_rad": -1.782815,
    "drift_mean": -0.00172777,
    "drift_var": 0.0012395162,
    "opacity_var": 0.001542461,
    "photosphere_crossed": false
  },
  {
    "step": 2284,
    "epoch": 5,
    "batch_in_epoch": 411,
    "theta1": 0.004389,
    "theta2": 0.00922,
    "loss": 0.027685,
    "eta2d_x": 0.397071,
    "eta2d_y": 0.917788,
    "speed": 3.2e-05,
    "heading_rad": -1.363498,
    "drift_mean": 0.00129458,
    "drift_var": 0.0014032459,
    "opacity_var": 0.0014202422,
    "photosphere_crossed": false
  },
  {
    "step": 2288,
    "epoch": 5,
    "batch_in_epoch": 415,
    "theta1": 0.004469,
    "theta2": 0.00913,
    "loss": 0.080033,
    "eta2d_x": 0.437354,
    "eta2d_y": 0.89929,
    "speed": 3.7e-05,
    "heading_rad": -0.778684,
    "drift_mean": 0.01308699,
    "drift_var": 0.0008516856,
    "opacity_var": 0.0010333165,
    "photosphere_crossed": false
  },
  {
    "step": 2292,
    "epoch": 5,
    "batch_in_epoch": 419,
    "theta1": 0.004551,
    "theta2": 0.009088,
    "loss": 0.071371,
    "eta2d_x": 0.470681,
    "eta2d_y": 0.882304,
    "speed": 1e-06,
    "heading_rad": 1.457071,
    "drift_mean": -0.00216558,
    "drift_var": 0.0016015772,
    "opacity_var": 0.0012619092,
    "photosphere_crossed": false
  },
  {
    "step": 2296,
    "epoch": 5,
    "batch_in_epoch": 423,
    "theta1": 0.004611,
    "theta2": 0.00917,
    "loss": 0.040819,
    "eta2d_x": 0.475688,
    "eta2d_y": 0.879614,
    "speed": 4.5e-05,
    "heading_rad": 0.813904,
    "drift_mean": -0.00763794,
    "drift_var": 0.0032826516,
    "opacity_var": 0.0024805259,
    "photosphere_crossed": false
  },
  {
    "step": 2300,
    "epoch": 5,
    "batch_in_epoch": 427,
    "theta1": 0.004701,
    "theta2": 0.009327,
    "loss": 0.039041,
    "eta2d_x": 0.477134,
    "eta2d_y": 0.878831,
    "speed": 4.2e-05,
    "heading_rad": 1.191253,
    "drift_mean": -0.00044446,
    "drift_var": 0.0004190731,
    "opacity_var": 0.0018698649,
    "photosphere_crossed": false
  },
  {
    "step": 2304,
    "epoch": 5,
    "batch_in_epoch": 431,
    "theta1": 0.004697,
    "theta2": 0.009433,
    "loss": 0.01904,
    "eta2d_x": 0.460068,
    "eta2d_y": 0.887884,
    "speed": 2.4e-05,
    "heading_rad": 2.018106,
    "drift_mean": -0.00500037,
    "drift_var": 0.0001551237,
    "opacity_var": 0.0003734719,
    "photosphere_crossed": false
  },
  {
    "step": 2308,
    "epoch": 5,
    "batch_in_epoch": 435,
    "theta1": 0.004646,
    "theta2": 0.009366,
    "loss": 0.073045,
    "eta2d_x": 0.455642,
    "eta2d_y": 0.890163,
    "speed": 1.4e-05,
    "heading_rad": -1.636834,
    "drift_mean": 0.01350121,
    "drift_var": 0.0013573128,
    "opacity_var": 0.0008719882,
    "photosphere_crossed": false
  },
  {
    "step": 2312,
    "epoch": 5,
    "batch_in_epoch": 439,
    "theta1": 0.004657,
    "theta2": 0.009347,
    "loss": 0.028932,
    "eta2d_x": 0.461496,
    "eta2d_y": 0.887142,
    "speed": 1.2e-05,
    "heading_rad": 2.776423,
    "drift_mean": -0.01102827,
    "drift_var": 0.0015204628,
    "opacity_var": 0.001416982,
    "photosphere_crossed": false
  },
  {
    "step": 2316,
    "epoch": 5,
    "batch_in_epoch": 443,
    "theta1": 0.004613,
    "theta2": 0.009402,
    "loss": 0.109489,
    "eta2d_x": 0.44097,
    "eta2d_y": 0.897522,
    "speed": 1.7e-05,
    "heading_rad": 2.038852,
    "drift_mean": 0.02013943,
    "drift_var": 0.0023983495,
    "opacity_var": 0.0021810108,
    "photosphere_crossed": false
  },
  {
    "step": 2320,
    "epoch": 5,
    "batch_in_epoch": 447,
    "theta1": 0.004496,
    "theta2": 0.009421,
    "loss": 0.078234,
    "eta2d_x": 0.403196,
    "eta2d_y": 0.915114,
    "speed": 2.9e-05,
    "heading_rad": 2.996878,
    "drift_mean": -0.00781381,
    "drift_var": 0.0019521612,
    "opacity_var": 0.002517082,
    "photosphere_crossed": false
  },
  {
    "step": 2324,
    "epoch": 5,
    "batch_in_epoch": 451,
    "theta1": 0.004421,
    "theta2": 0.009388,
    "loss": 0.01569,
    "eta2d_x": 0.384214,
    "eta2d_y": 0.923244,
    "speed": 1.5e-05,
    "heading_rad": -2.136682,
    "drift_mean": -0.01563613,
    "drift_var": 0.0001995972,
    "opacity_var": 0.0014007314,
    "photosphere_crossed": false
  },
  {
    "step": 2328,
    "epoch": 5,
    "batch_in_epoch": 455,
    "theta1": 0.004461,
    "theta2": 0.009266,
    "loss": 0.110652,
    "eta2d_x": 0.41408,
    "eta2d_y": 0.910241,
    "speed": 6.6e-05,
    "heading_rad": -0.914709,
    "drift_mean": 0.02374056,
    "drift_var": 0.0007524168,
    "opacity_var": 0.0008011792,
    "photosphere_crossed": false
  },
  {
    "step": 2332,
    "epoch": 5,
    "batch_in_epoch": 459,
    "theta1": 0.00454,
    "theta2": 0.009124,
    "loss": 0.033548,
    "eta2d_x": 0.461292,
    "eta2d_y": 0.887248,
    "speed": 3.2e-05,
    "heading_rad": -1.14206,
    "drift_mean": -0.01927604,
    "drift_var": 0.0042253175,
    "opacity_var": 0.0027217647,
    "photosphere_crossed": false
  },
  {
    "step": 2336,
    "epoch": 5,
    "batch_in_epoch": 463,
    "theta1": 0.004623,
    "theta2": 0.009059,
    "loss": 0.027144,
    "eta2d_x": 0.498089,
    "eta2d_y": 0.867126,
    "speed": 3.1e-05,
    "heading_rad": -0.148291,
    "drift_mean": -0.00160084,
    "drift_var": 0.0002727733,
    "opacity_var": 0.0025709051,
    "photosphere_crossed": false
  },
  {
    "step": 2340,
    "epoch": 5,
    "batch_in_epoch": 467,
    "theta1": 0.004726,
    "theta2": 0.009035,
    "loss": 0.00796,
    "eta2d_x": 0.532844,
    "eta2d_y": 0.846214,
    "speed": 2.7e-05,
    "heading_rad": -0.173096,
    "drift_mean": -0.00479608,
    "drift_var": 0.0017989099,
    "opacity_var": 0.0010968269,
    "photosphere_crossed": false
  },
  {
    "step": 2344,
    "epoch": 6,
    "batch_in_epoch": 3,
    "theta1": 0.004711,
    "theta2": 0.009029,
    "loss": 0.075288,
    "eta2d_x": 0.529821,
    "eta2d_y": 0.84811,
    "speed": 2.3e-05,
    "heading_rad": -2.894865,
    "drift_mean": 0.01683203,
    "drift_var": 0.0025528265,
    "opacity_var": 0.0020553072,
    "photosphere_crossed": false
  },
  {
    "step": 2348,
    "epoch": 6,
    "batch_in_epoch": 7,
    "theta1": 0.004623,
    "theta2": 0.008958,
    "loss": 0.040545,
    "eta2d_x": 0.516289,
    "eta2d_y": 0.856414,
    "speed": 5e-05,
    "heading_rad": -2.333857,
    "drift_mean": -0.00868574,
    "drift_var": 0.0012681578,
    "opacity_var": 0.0022436701,
    "photosphere_crossed": false
  },
  {
    "step": 2352,
    "epoch": 6,
    "batch_in_epoch": 11,
    "theta1": 0.004527,
    "theta2": 0.008794,
    "loss": 0.018085,
    "eta2d_x": 0.517406,
    "eta2d_y": 0.85574,
    "speed": 5.2e-05,
    "heading_rad": -1.769921,
    "drift_mean": -0.0056151,
    "drift_var": 0.0003332822,
    "opacity_var": 0.0012195846,
    "photosphere_crossed": false
  },
  {
    "step": 2356,
    "epoch": 6,
    "batch_in_epoch": 15,
    "theta1": 0.004531,
    "theta2": 0.008674,
    "loss": 0.021961,
    "eta2d_x": 0.543865,
    "eta2d_y": 0.839173,
    "speed": 1.5e-05,
    "heading_rad": -2.414623,
    "drift_mean": 0.00096904,
    "drift_var": 0.0001896912,
    "opacity_var": 0.0002422003,
    "photosphere_crossed": true
  },
  {
    "step": 2360,
    "epoch": 6,
    "batch_in_epoch": 19,
    "theta1": 0.004458,
    "theta2": 0.008609,
    "loss": 0.04357,
    "eta2d_x": 0.53299,
    "eta2d_y": 0.846122,
    "speed": 4e-05,
    "heading_rad": -2.509134,
    "drift_mean": 0.0054022,
    "drift_var": 0.0002801713,
    "opacity_var": 0.0003011664,
    "photosphere_crossed": false
  },
  {
    "step": 2364,
    "epoch": 6,
    "batch_in_epoch": 23,
    "theta1": 0.0044,
    "theta2": 0.008511,
    "loss": 0.01075,
    "eta2d_x": 0.534253,
    "eta2d_y": 0.845325,
    "speed": 1.6e-05,
    "heading_rad": -1.409628,
    "drift_mean": -0.00820499,
    "drift_var": 0.00063434,
    "opacity_var": 0.0004661719,
    "photosphere_crossed": false
  },
  {
    "step": 2368,
    "epoch": 6,
    "batch_in_epoch": 27,
    "theta1": 0.004384,
    "theta2": 0.008462,
    "loss": 0.009946,
    "eta2d_x": 0.54006,
    "eta2d_y": 0.841627,
    "speed": 1.2e-05,
    "heading_rad": -1.962975,
    "drift_mean": -0.00020086,
    "drift_var": 0.0001489242,
    "opacity_var": 0.0003795088,
    "photosphere_crossed": false
  },
  {
    "step": 2372,
    "epoch": 6,
    "batch_in_epoch": 31,
    "theta1": 0.004454,
    "theta2": 0.008475,
    "loss": 0.022294,
    "eta2d_x": 0.563169,
    "eta2d_y": 0.826342,
    "speed": 2.5e-05,
    "heading_rad": 0.745123,
    "drift_mean": 0.00308686,
    "drift_var": 8.69603e-05,
    "opacity_var": 0.0001698034,
    "photosphere_crossed": true
  },
  {
    "step": 2376,
    "epoch": 6,
    "batch_in_epoch": 35,
    "theta1": 0.004472,
    "theta2": 0.00849,
    "loss": 0.022492,
    "eta2d_x": 0.566102,
    "eta2d_y": 0.824335,
    "speed": 1.2e-05,
    "heading_rad": -1.186077,
    "drift_mean": 4.951e-05,
    "drift_var": 0.0005394581,
    "opacity_var": 0.0002912565,
    "photosphere_crossed": true
  },
  {
    "step": 2380,
    "epoch": 6,
    "batch_in_epoch": 39,
    "theta1": 0.004446,
    "theta2": 0.008309,
    "loss": 0.012052,
    "eta2d_x": 0.603877,
    "eta2d_y": 0.797078,
    "speed": 3.8e-05,
    "heading_rad": -1.622284,
    "drift_mean": -0.00261,
    "drift_var": 0.0006836117,
    "opacity_var": 0.000550396,
    "photosphere_crossed": false
  },
  {
    "step": 2384,
    "epoch": 6,
    "batch_in_epoch": 43,
    "theta1": 0.004426,
    "theta2": 0.008177,
    "loss": 0.028225,
    "eta2d_x": 0.634378,
    "eta2d_y": 0.773023,
    "speed": 3.9e-05,
    "heading_rad": -1.965197,
    "drift_mean": 0.00404322,
    "drift_var": 3.71755e-05,
    "opacity_var": 0.0003393282,
    "photosphere_crossed": false
  },
  {
    "step": 2388,
    "epoch": 6,
    "batch_in_epoch": 47,
    "theta1": 0.004431,
    "theta2": 0.008144,
    "loss": 0.051547,
    "eta2d_x": 0.646207,
    "eta2d_y": 0.763163,
    "speed": 8e-06,
    "heading_rad": 0.94153,
    "drift_mean": 0.00583051,
    "drift_var": 0.0003852941,
    "opacity_var": 0.0002292773,
    "photosphere_crossed": true
  },
  {
    "step": 2392,
    "epoch": 6,
    "batch_in_epoch": 51,
    "theta1": 0.004449,
    "theta2": 0.008147,
    "loss": 0.010444,
    "eta2d_x": 0.651967,
    "eta2d_y": 0.758247,
    "speed": 6e-06,
    "heading_rad": 1.85205,
    "drift_mean": -0.01027566,
    "drift_var": 0.0003701533,
    "opacity_var": 0.0003949605,
    "photosphere_crossed": false
  },
  {
    "step": 2396,
    "epoch": 6,
    "batch_in_epoch": 55,
    "theta1": 0.004396,
    "theta2": 0.008183,
    "loss": 0.021054,
    "eta2d_x": 0.621557,
    "eta2d_y": 0.783369,
    "speed": 2e-06,
    "heading_rad": -1.801078,
    "drift_mean": 0.00265251,
    "drift_var": 8.2969e-05,
    "opacity_var": 0.0003525505,
    "photosphere_crossed": false
  },
  {
    "step": 2400,
    "epoch": 6,
    "batch_in_epoch": 59,
    "theta1": 0.004374,
    "theta2": 0.008139,
    "loss": 0.046935,
    "eta2d_x": 0.626698,
    "eta2d_y": 0.779262,
    "speed": 1.2e-05,
    "heading_rad": -1.668356,
    "drift_mean": 0.0064702,
    "drift_var": 7.56403e-05,
    "opacity_var": 0.0001072305,
    "photosphere_crossed": true
  },
  {
    "step": 2404,
    "epoch": 6,
    "batch_in_epoch": 63,
    "theta1": 0.004374,
    "theta2": 0.008004,
    "loss": 0.037519,
    "eta2d_x": 0.670699,
    "eta2d_y": 0.74173,
    "speed": 4e-05,
    "heading_rad": -1.34791,
    "drift_mean": -0.00235403,
    "drift_var": 0.0003225459,
    "opacity_var": 0.0001964093,
    "photosphere_crossed": true
  },
  {
    "step": 2408,
    "epoch": 6,
    "batch_in_epoch": 67,
    "theta1": 0.004532,
    "theta2": 0.00797,
    "loss": 0.004708,
    "eta2d_x": 0.734855,
    "eta2d_y": 0.678224,
    "speed": 4.2e-05,
    "heading_rad": 0.047397,
    "drift_mean": -0.00820271,
    "drift_var": 0.0009933904,
    "opacity_var": 0.0006449857,
    "photosphere_crossed": false
  },
  {
    "step": 2412,
    "epoch": 6,
    "batch_in_epoch": 71,
    "theta1": 0.004741,
    "theta2": 0.008024,
    "loss": 0.018872,
    "eta2d_x": 0.773454,
    "eta2d_y": 0.633852,
    "speed": 5.1e-05,
    "heading_rad": 0.498484,
    "drift_mean": 0.00354091,
    "drift_var": 0.0082017733,
    "opacity_var": 0.0041722201,
    "photosphere_crossed": false
  },
  {
    "step": 2416,
    "epoch": 6,
    "batch_in_epoch": 75,
    "theta1": 0.004896,
    "theta2": 0.008143,
    "loss": 0.020783,
    "eta2d_x": 0.775387,
    "eta2d_y": 0.631486,
    "speed": 4.6e-05,
    "heading_rad": 0.642683,
    "drift_mean": 0.0004778,
    "drift_var": 0.0003996574,
    "opacity_var": 0.0038886767,
    "photosphere_crossed": false
  },
  {
    "step": 2420,
    "epoch": 6,
    "batch_in_epoch": 79,
    "theta1": 0.004927,
    "theta2": 0.008203,
    "loss": 0.02554,
    "eta2d_x": 0.766805,
    "eta2d_y": 0.64188,
    "speed": 9e-06,
    "heading_rad": -2.846078,
    "drift_mean": 0.00118926,
    "drift_var": 6.79839e-05,
    "opacity_var": 0.0002187131,
    "photosphere_crossed": true
  },
  {
    "step": 2424,
    "epoch": 6,
    "batch_in_epoch": 83,
    "theta1": 0.004905,
    "theta2": 0.008215,
    "loss": 0.016158,
    "eta2d_x": 0.75936,
    "eta2d_y": 0.650671,
    "speed": 1.5e-05,
    "heading_rad": 0.939346,
    "drift_mean": -0.00234538,
    "drift_var": 0.0002570327,
    "opacity_var": 0.0001475675,
    "photosphere_crossed": true
  },
  {
    "step": 2428,
    "epoch": 6,
    "batch_in_epoch": 87,
    "theta1": 0.004871,
    "theta2": 0.008192,
    "loss": 0.015724,
    "eta2d_x": 0.757895,
    "eta2d_y": 0.652377,
    "speed": 1.3e-05,
    "heading_rad": -2.801283,
    "drift_mean": -0.00010853,
    "drift_var": 0.0010861628,
    "opacity_var": 0.0005982413,
    "photosphere_crossed": false
  },
  {
    "step": 2432,
    "epoch": 6,
    "batch_in_epoch": 91,
    "theta1": 0.004813,
    "theta2": 0.008164,
    "loss": 0.093771,
    "eta2d_x": 0.752049,
    "eta2d_y": 0.659107,
    "speed": 7e-06,
    "heading_rad": -2.418369,
    "drift_mean": 0.01951167,
    "drift_var": 0.0003248602,
    "opacity_var": 0.0007871524,
    "photosphere_crossed": false
  },
  {
    "step": 2436,
    "epoch": 6,
    "batch_in_epoch": 95,
    "theta1": 0.004847,
    "theta2": 0.00823,
    "loss": 0.019484,
    "eta2d_x": 0.742502,
    "eta2d_y": 0.669844,
    "speed": 1.8e-05,
    "heading_rad": 1.133284,
    "drift_mean": -0.0185717,
    "drift_var": 0.0010987176,
    "opacity_var": 0.0009550076,
    "photosphere_crossed": false
  },
  {
    "step": 2440,
    "epoch": 6,
    "batch_in_epoch": 99,
    "theta1": 0.004855,
    "theta2": 0.008266,
    "loss": 0.039015,
    "eta2d_x": 0.735317,
    "eta2d_y": 0.677723,
    "speed": 3e-06,
    "heading_rad": -2.580425,
    "drift_mean": 0.00488271,
    "drift_var": 0.0002440985,
    "opacity_var": 0.0008446611,
    "photosphere_crossed": false
  },
  {
    "step": 2444,
    "epoch": 6,
    "batch_in_epoch": 103,
    "theta1": 0.004813,
    "theta2": 0.008288,
    "loss": 0.032819,
    "eta2d_x": 0.719763,
    "eta2d_y": 0.69422,
    "speed": 9e-06,
    "heading_rad": 1.55639,
    "drift_mean": -0.00154885,
    "drift_var": 0.0003400371,
    "opacity_var": 0.0002792155,
    "photosphere_crossed": true
  },
  {
    "step": 2448,
    "epoch": 6,
    "batch_in_epoch": 107,
    "theta1": 0.004836,
    "theta2": 0.008308,
    "loss": 0.018595,
    "eta2d_x": 0.720404,
    "eta2d_y": 0.693554,
    "speed": 1e-05,
    "heading_rad": 0.242173,
    "drift_mean": -0.00355608,
    "drift_var": 0.0005301911,
    "opacity_var": 0.0003902149,
    "photosphere_crossed": false
  },
  {
    "step": 2452,
    "epoch": 6,
    "batch_in_epoch": 111,
    "theta1": 0.0049,
    "theta2": 0.008299,
    "loss": 0.011582,
    "eta2d_x": 0.737391,
    "eta2d_y": 0.675467,
    "speed": 1.5e-05,
    "heading_rad": -0.313845,
    "drift_mean": -0.00175332,
    "drift_var": 0.0010056496,
    "opacity_var": 0.0007640986,
    "photosphere_crossed": false
  },
  {
    "step": 2456,
    "epoch": 6,
    "batch_in_epoch": 115,
    "theta1": 0.005033,
    "theta2": 0.008301,
    "loss": 0.031461,
    "eta2d_x": 0.764528,
    "eta2d_y": 0.64459,
    "speed": 3.9e-05,
    "heading_rad": -0.070943,
    "drift_mean": 0.00496975,
    "drift_var": 0.00044849,
    "opacity_var": 0.0007176059,
    "photosphere_crossed": false
  },
  {
    "step": 2460,
    "epoch": 6,
    "batch_in_epoch": 119,
    "theta1": 0.005165,
    "theta2": 0.008265,
    "loss": 0.019994,
    "eta2d_x": 0.796356,
    "eta2d_y": 0.604828,
    "speed": 3.6e-05,
    "heading_rad": -0.312133,
    "drift_mean": -0.00286668,
    "drift_var": 0.0002181971,
    "opacity_var": 0.000310854,
    "photosphere_crossed": false
  },
  {
    "step": 2464,
    "epoch": 6,
    "batch_in_epoch": 123,
    "theta1": 0.005335,
    "theta2": 0.008313,
    "loss": 0.02874,
    "eta2d_x": 0.812403,
    "eta2d_y": 0.583096,
    "speed": 5e-05,
    "heading_rad": 0.046795,
    "drift_mean": 0.0021865,
    "drift_var": 0.0005457253,
    "opacity_var": 0.0003600216,
    "photosphere_crossed": false
  },
  {
    "step": 2468,
    "epoch": 6,
    "batch_in_epoch": 127,
    "theta1": 0.00544,
    "theta2": 0.008308,
    "loss": 0.010636,
    "eta2d_x": 0.827403,
    "eta2d_y": 0.561609,
    "speed": 1.3e-05,
    "heading_rad": -0.296351,
    "drift_mean": -0.00452602,
    "drift_var": 5.40862e-05,
    "opacity_var": 0.0002772263,
    "photosphere_crossed": true
  },
  {
    "step": 2472,
    "epoch": 6,
    "batch_in_epoch": 131,
    "theta1": 0.005339,
    "theta2": 0.008292,
    "loss": 0.010099,
    "eta2d_x": 0.817114,
    "eta2d_y": 0.576476,
    "speed": 4.6e-05,
    "heading_rad": -2.894331,
    "drift_mean": -0.00013431,
    "drift_var": 0.0009876164,
    "opacity_var": 0.0005216541,
    "photosphere_crossed": false
  },
  {
    "step": 2476,
    "epoch": 6,
    "batch_in_epoch": 135,
    "theta1": 0.005174,
    "theta2": 0.008269,
    "loss": 0.025973,
    "eta2d_x": 0.796866,
    "eta2d_y": 0.604156,
    "speed": 4.4e-05,
    "heading_rad": 3.065348,
    "drift_mean": 0.00396853,
    "drift_var": 0.0001100308,
    "opacity_var": 0.0005229081,
    "photosphere_crossed": false
  },
  {
    "step": 2480,
    "epoch": 6,
    "batch_in_epoch": 139,
    "theta1": 0.005063,
    "theta2": 0.008283,
    "loss": 0.016295,
    "eta2d_x": 0.77455,
    "eta2d_y": 0.632513,
    "speed": 2.1e-05,
    "heading_rad": 2.839074,
    "drift_mean": -0.00241948,
    "drift_var": 0.0004139125,
    "opacity_var": 0.0003007269,
    "photosphere_crossed": false
  },
  {
    "step": 2484,
    "epoch": 6,
    "batch_in_epoch": 143,
    "theta1": 0.005019,
    "theta2": 0.00834,
    "loss": 0.004831,
    "eta2d_x": 0.752937,
    "eta2d_y": 0.658092,
    "speed": 1.9e-05,
    "heading_rad": 2.41483,
    "drift_mean": -0.00286603,
    "drift_var": 0.0001105026,
    "opacity_var": 0.0002459374,
    "photosphere_crossed": true
  },
  {
    "step": 2488,
    "epoch": 6,
    "batch_in_epoch": 147,
    "theta1": 0.004969,
    "theta2": 0.008402,
    "loss": 0.035461,
    "eta2d_x": 0.728073,
    "eta2d_y": 0.6855,
    "speed": 1.8e-05,
    "heading_rad": 2.431299,
    "drift_mean": 0.00765762,
    "drift_var": 0.0007940702,
    "opacity_var": 0.0004287791,
    "photosphere_crossed": false
  },
  {
    "step": 2492,
    "epoch": 6,
    "batch_in_epoch": 151,
    "theta1": 0.004901,
    "theta2": 0.008509,
    "loss": 0.047086,
    "eta2d_x": 0.687943,
    "eta2d_y": 0.725765,
    "speed": 4.1e-05,
    "heading_rad": 1.747657,
    "drift_mean": 0.00290625,
    "drift_var": 0.0008519029,
    "opacity_var": 0.0007791559,
    "photosphere_crossed": false
  },
  {
    "step": 2496,
    "epoch": 6,
    "batch_in_epoch": 155,
    "theta1": 0.004933,
    "theta2": 0.008632,
    "loss": 0.042152,
    "eta2d_x": 0.668501,
    "eta2d_y": 0.743712,
    "speed": 3.2e-05,
    "heading_rad": 0.988075,
    "drift_mean": -0.00123368,
    "drift_var": 0.0004245847,
    "opacity_var": 0.0006510384,
    "photosphere_crossed": false
  },
  {
    "step": 2500,
    "epoch": 6,
    "batch_in_epoch": 159,
    "theta1": 0.004983,
    "theta2": 0.008651,
    "loss": 0.03073,
    "eta2d_x": 0.676244,
    "eta2d_y": 0.736677,
    "speed": 1e-05,
    "heading_rad": 0.224227,
    "drift_mean": -0.00285549,
    "drift_var": 0.0025254459,
    "opacity_var": 0.0014377279,
    "photosphere_crossed": false
  },
  {
    "step": 2504,
    "epoch": 6,
    "batch_in_epoch": 163,
    "theta1": 0.00503,
    "theta2": 0.008589,
    "loss": 0.020838,
    "eta2d_x": 0.699904,
    "eta2d_y": 0.714237,
    "speed": 3.7e-05,
    "heading_rad": -1.3433,
    "drift_mean": -0.00247282,
    "drift_var": 6.84037e-05,
    "opacity_var": 0.0012806969,
    "photosphere_crossed": false
  },
  {
    "step": 2508,
    "epoch": 6,
    "batch_in_epoch": 167,
    "theta1": 0.004975,
    "theta2": 0.008402,
    "loss": 0.147365,
    "eta2d_x": 0.729603,
    "eta2d_y": 0.683871,
    "speed": 5.6e-05,
    "heading_rad": -2.013961,
    "drift_mean": 0.03163164,
    "drift_var": 0.0034662753,
    "opacity_var": 0.0018329311,
    "photosphere_crossed": false
  },
  {
    "step": 2512,
    "epoch": 6,
    "batch_in_epoch": 171,
    "theta1": 0.004918,
    "theta2": 0.008217,
    "loss": 0.028788,
    "eta2d_x": 0.761405,
    "eta2d_y": 0.648277,
    "speed": 5e-05,
    "heading_rad": -1.944542,
    "drift_mean": -0.02964425,
    "drift_var": 0.0014578584,
    "opacity_var": 0.0030248376,
    "photosphere_crossed": false
  },
  {
    "step": 2516,
    "epoch": 6,
    "batch_in_epoch": 175,
    "theta1": 0.004882,
    "theta2": 0.008077,
    "loss": 0.065315,
    "eta2d_x": 0.789421,
    "eta2d_y": 0.613852,
    "speed": 3.7e-05,
    "heading_rad": -1.603768,
    "drift_mean": 0.00913182,
    "drift_var": 0.0012631942,
    "opacity_var": 0.0027320906,
    "photosphere_crossed": false
  },
  {
    "step": 2520,
    "epoch": 6,
    "batch_in_epoch": 179,
    "theta1": 0.004883,
    "theta2": 0.008006,
    "loss": 0.009671,
    "eta2d_x": 0.80774,
    "eta2d_y": 0.589539,
    "speed": 1.6e-05,
    "heading_rad": -0.995582,
    "drift_mean": -0.01391118,
    "drift_var": 0.0001863228,
    "opacity_var": 0.0007690539,
    "photosphere_crossed": false
  },
  {
    "step": 2524,
    "epoch": 6,
    "batch_in_epoch": 183,
    "theta1": 0.004731,
    "theta2": 0.00794,
    "loss": 0.049977,
    "eta2d_x": 0.795016,
    "eta2d_y": 0.606588,
    "speed": 5.6e-05,
    "heading_rad": -2.797818,
    "drift_mean": 0.01007672,
    "drift_var": 0.0011402631,
    "opacity_var": 0.0010473475,
    "photosphere_crossed": false
  },
  {
    "step": 2528,
    "epoch": 6,
    "batch_in_epoch": 187,
    "theta1": 0.004602,
    "theta2": 0.007888,
    "loss": 0.082183,
    "eta2d_x": 0.78024,
    "eta2d_y": 0.62548,
    "speed": 2.5e-05,
    "heading_rad": -3.133464,
    "drift_mean": 0.00805147,
    "drift_var": 0.0021250699,
    "opacity_var": 0.0015049261,
    "photosphere_crossed": false
  },
  {
    "step": 2532,
    "epoch": 6,
    "batch_in_epoch": 191,
    "theta1": 0.004572,
    "theta2": 0.007925,
    "loss": 0.03494,
    "eta2d_x": 0.760539,
    "eta2d_y": 0.649292,
    "speed": 9e-06,
    "heading_rad": -2.783998,
    "drift_mean": -0.01181091,
    "drift_var": 0.00171745,
    "opacity_var": 0.0019309373,
    "photosphere_crossed": false
  },
  {
    "step": 2536,
    "epoch": 6,
    "batch_in_epoch": 195,
    "theta1": 0.004471,
    "theta2": 0.007938,
    "loss": 0.025219,
    "eta2d_x": 0.726464,
    "eta2d_y": 0.687205,
    "speed": 2.3e-05,
    "heading_rad": -2.887023,
    "drift_mean": -0.00243019,
    "drift_var": 0.0001171476,
    "opacity_var": 0.0015279815,
    "photosphere_crossed": false
  },
  {
    "step": 2540,
    "epoch": 6,
    "batch_in_epoch": 199,
    "theta1": 0.004517,
    "theta2": 0.007888,
    "loss": 0.010727,
    "eta2d_x": 0.757438,
    "eta2d_y": 0.652907,
    "speed": 1.6e-05,
    "heading_rad": -0.73225,
    "drift_mean": -0.00362298,
    "drift_var": 0.00083584,
    "opacity_var": 0.0004244166,
    "photosphere_crossed": false
  },
  {
    "step": 2544,
    "epoch": 6,
    "batch_in_epoch": 203,
    "theta1": 0.004567,
    "theta2": 0.007971,
    "loss": 0.033845,
    "eta2d_x": 0.744702,
    "eta2d_y": 0.667397,
    "speed": 5.8e-05,
    "heading_rad": 1.179848,
    "drift_mean": 0.00577953,
    "drift_var": 0.0004549295,
    "opacity_var": 0.0005937963,
    "photosphere_crossed": false
  },
  {
    "step": 2548,
    "epoch": 6,
    "batch_in_epoch": 207,
    "theta1": 0.004678,
    "theta2": 0.008167,
    "loss": 0.013002,
    "eta2d_x": 0.717183,
    "eta2d_y": 0.696885,
    "speed": 5.8e-05,
    "heading_rad": 1.040868,
    "drift_mean": -0.00521086,
    "drift_var": 0.0047976367,
    "opacity_var": 0.0025977017,
    "photosphere_crossed": false
  },
  {
    "step": 2552,
    "epoch": 6,
    "batch_in_epoch": 211,
    "theta1": 0.0047,
    "theta2": 0.008377,
    "loss": 0.061706,
    "eta2d_x": 0.66749,
    "eta2d_y": 0.744619,
    "speed": 4.2e-05,
    "heading_rad": 2.630628,
    "drift_mean": 0.01217618,
    "drift_var": 0.0005964921,
    "opacity_var": 0.0025595439,
    "photosphere_crossed": false
  },
  {
    "step": 2556,
    "epoch": 6,
    "batch_in_epoch": 215,
    "theta1": 0.00455,
    "theta2": 0.008407,
    "loss": 0.029498,
    "eta2d_x": 0.613583,
    "eta2d_y": 0.78963,
    "speed": 3.3e-05,
    "heading_rad": 3.048179,
    "drift_mean": -0.00805211,
    "drift_var": 0.0002609973,
    "opacity_var": 0.0014701905,
    "photosphere_crossed": false
  },
  {
    "step": 2560,
    "epoch": 6,
    "batch_in_epoch": 219,
    "theta1": 0.00454,
    "theta2": 0.008305,
    "loss": 0.013867,
    "eta2d_x": 0.637419,
    "eta2d_y": 0.770517,
    "speed": 2.6e-05,
    "heading_rad": -1.542537,
    "drift_mean": -0.00390763,
    "drift_var": 0.0023967745,
    "opacity_var": 0.0012824989,
    "photosphere_crossed": false
  },
  {
    "step": 2564,
    "epoch": 6,
    "batch_in_epoch": 223,
    "theta1": 0.004551,
    "theta2": 0.008335,
    "loss": 0.033304,
    "eta2d_x": 0.632819,
    "eta2d_y": 0.774299,
    "speed": 2.7e-05,
    "heading_rad": 0.686915,
    "drift_mean": 0.00485914,
    "drift_var": 0.0002027725,
    "opacity_var": 0.0011753923,
    "photosphere_crossed": false
  },
  {
    "step": 2568,
    "epoch": 6,
    "batch_in_epoch": 227,
    "theta1": 0.004799,
    "theta2": 0.008327,
    "loss": 0.134111,
    "eta2d_x": 0.706414,
    "eta2d_y": 0.707799,
    "speed": 7.8e-05,
    "heading_rad": 0.122883,
    "drift_mean": 0.02520186,
    "drift_var": 0.0023783184,
    "opacity_var": 0.0012598949,
    "photosphere_crossed": false
  },
  {
    "step": 2572,
    "epoch": 6,
    "batch_in_epoch": 231,
    "theta1": 0.00499,
    "theta2": 0.008425,
    "loss": 0.068043,
    "eta2d_x": 0.727505,
    "eta2d_y": 0.686102,
    "speed": 2.8e-05,
    "heading_rad": 1.111727,
    "drift_mean": -0.01651721,
    "drift_var": 0.0055721291,
    "opacity_var": 0.0039468305,
    "photosphere_crossed": false
  },
  {
    "step": 2576,
    "epoch": 6,
    "batch_in_epoch": 235,
    "theta1": 0.005026,
    "theta2": 0.008494,
    "loss": 0.041094,
    "eta2d_x": 0.719761,
    "eta2d_y": 0.694222,
    "speed": 9e-06,
    "heading_rad": 1.212533,
    "drift_mean": -0.00673714,
    "drift_var": 0.0038851269,
    "opacity_var": 0.0054636009,
    "photosphere_crossed": false
  },
  {
    "step": 2580,
    "epoch": 6,
    "batch_in_epoch": 239,
    "theta1": 0.005025,
    "theta2": 0.00862,
    "loss": 0.058854,
    "eta2d_x": 0.692403,
    "eta2d_y": 0.721511,
    "speed": 6.3e-05,
    "heading_rad": 1.503764,
    "drift_mean": 0.00443995,
    "drift_var": 0.0003712394,
    "opacity_var": 0.0019959966,
    "photosphere_crossed": false
  },
  {
    "step": 2584,
    "epoch": 6,
    "batch_in_epoch": 243,
    "theta1": 0.005009,
    "theta2": 0.008777,
    "loss": 0.040857,
    "eta2d_x": 0.65636,
    "eta2d_y": 0.754448,
    "speed": 2.9e-05,
    "heading_rad": 1.868027,
    "drift_mean": -0.00449913,
    "drift_var": 0.000843439,
    "opacity_var": 0.0010040122,
    "photosphere_crossed": false
  },
  {
    "step": 2588,
    "epoch": 6,
    "batch_in_epoch": 247,
    "theta1": 0.005025,
    "theta2": 0.008846,
    "loss": 0.067127,
    "eta2d_x": 0.646348,
    "eta2d_y": 0.763043,
    "speed": 1.7e-05,
    "heading_rad": 0.471412,
    "drift_mean": 0.00656752,
    "drift_var": 0.0016533116,
    "opacity_var": 0.0011914449,
    "photosphere_crossed": false
  },
  {
    "step": 2592,
    "epoch": 6,
    "batch_in_epoch": 251,
    "theta1": 0.005047,
    "theta2": 0.008856,
    "loss": 0.019837,
    "eta2d_x": 0.649266,
    "eta2d_y": 0.760561,
    "speed": 8e-06,
    "heading_rad": -0.679925,
    "drift_mean": -0.01182253,
    "drift_var": 0.0004075492,
    "opacity_var": 0.0010101786,
    "photosphere_crossed": false
  },
  {
    "step": 2596,
    "epoch": 6,
    "batch_in_epoch": 255,
    "theta1": 0.005008,
    "theta2": 0.008841,
    "loss": 0.015203,
    "eta2d_x": 0.643226,
    "eta2d_y": 0.765677,
    "speed": 7e-06,
    "heading_rad": 2.297815,
    "drift_mean": -0.00115856,
    "drift_var": 0.0001721686,
    "opacity_var": 0.0004602712,
    "photosphere_crossed": false
  },
  {
    "step": 2600,
    "epoch": 6,
    "batch_in_epoch": 259,
    "theta1": 0.004876,
    "theta2": 0.00885,
    "loss": 0.017606,
    "eta2d_x": 0.608779,
    "eta2d_y": 0.79334,
    "speed": 4.1e-05,
    "heading_rad": 3.136489,
    "drift_mean": 0.00060072,
    "drift_var": 0.0006046103,
    "opacity_var": 0.0003460562,
    "photosphere_crossed": false
  },
  {
    "step": 2604,
    "epoch": 6,
    "batch_in_epoch": 263,
    "theta1": 0.004819,
    "theta2": 0.008873,
    "loss": 0.033031,
    "eta2d_x": 0.589326,
    "eta2d_y": 0.807895,
    "speed": 1.6e-05,
    "heading_rad": 3.076787,
    "drift_mean": 0.00385623,
    "drift_var": 0.0006765718,
    "opacity_var": 0.000590825,
    "photosphere_crossed": false
  },
  {
    "step": 2608,
    "epoch": 6,
    "batch_in_epoch": 267,
    "theta1": 0.004729,
    "theta2": 0.008823,
    "loss": 0.026653,
    "eta2d_x": 0.574211,
    "eta2d_y": 0.818707,
    "speed": 2.7e-05,
    "heading_rad": -2.815604,
    "drift_mean": -0.00159456,
    "drift_var": 0.0002197177,
    "opacity_var": 0.0004704507,
    "photosphere_crossed": false
  },
  {
    "step": 2612,
    "epoch": 6,
    "batch_in_epoch": 271,
    "theta1": 0.004698,
    "theta2": 0.008895,
    "loss": 0.055091,
    "eta2d_x": 0.550937,
    "eta2d_y": 0.834547,
    "speed": 2.9e-05,
    "heading_rad": 1.44548,
    "drift_mean": 0.00710963,
    "drift_var": 0.0032600179,
    "opacity_var": 0.0015990865,
    "photosphere_crossed": false
  },
  {
    "step": 2616,
    "epoch": 6,
    "batch_in_epoch": 275,
    "theta1": 0.004679,
    "theta2": 0.008954,
    "loss": 0.034567,
    "eta2d_x": 0.533979,
    "eta2d_y": 0.845498,
    "speed": 2.3e-05,
    "heading_rad": 1.960281,
    "drift_mean": -0.00513113,
    "drift_var": 0.0004887346,
    "opacity_var": 0.0017005004,
    "photosphere_crossed": false
  },
  {
    "step": 2620,
    "epoch": 6,
    "batch_in_epoch": 279,
    "theta1": 0.004583,
    "theta2": 0.008966,
    "loss": 0.014405,
    "eta2d_x": 0.502482,
    "eta2d_y": 0.864587,
    "speed": 2.4e-05,
    "heading_rad": -2.627774,
    "drift_mean": -0.00504032,
    "drift_var": 0.0007661308,
    "opacity_var": 0.000772141,
    "photosphere_crossed": false
  },
  {
    "step": 2624,
    "epoch": 6,
    "batch_in_epoch": 283,
    "theta1": 0.004516,
    "theta2": 0.008853,
    "loss": 0.036814,
    "eta2d_x": 0.501988,
    "eta2d_y": 0.864875,
    "speed": 2.8e-05,
    "heading_rad": -1.453349,
    "drift_mean": 0.0056022,
    "drift_var": 0.0050563617,
    "opacity_var": 0.002629288,
    "photosphere_crossed": false
  },
  {
    "step": 2628,
    "epoch": 6,
    "batch_in_epoch": 287,
    "theta1": 0.004563,
    "theta2": 0.008761,
    "loss": 0.026751,
    "eta2d_x": 0.535731,
    "eta2d_y": 0.844388,
    "speed": 3.2e-05,
    "heading_rad": -0.717942,
    "drift_mean": -0.00251568,
    "drift_var": 0.0004312118,
    "opacity_var": 0.0024672134,
    "photosphere_crossed": false
  },
  {
    "step": 2632,
    "epoch": 6,
    "batch_in_epoch": 291,
    "theta1": 0.004567,
    "theta2": 0.008587,
    "loss": 0.056478,
    "eta2d_x": 0.575402,
    "eta2d_y": 0.81787,
    "speed": 6.7e-05,
    "heading_rad": -1.893903,
    "drift_mean": 0.00743163,
    "drift_var": 6.41912e-05,
    "opacity_var": 0.0002984868,
    "photosphere_crossed": true
  },
  {
    "step": 2636,
    "epoch": 6,
    "batch_in_epoch": 295,
    "theta1": 0.004482,
    "theta2": 0.008409,
    "loss": 0.079609,
    "eta2d_x": 0.590045,
    "eta2d_y": 0.80737,
    "speed": 2.8e-05,
    "heading_rad": -2.041514,
    "drift_mean": 0.00578267,
    "drift_var": 0.0009933325,
    "opacity_var": 0.0005646138,
    "photosphere_crossed": false
  },
  {
    "step": 2640,
    "epoch": 6,
    "batch_in_epoch": 299,
    "theta1": 0.004412,
    "theta2": 0.008382,
    "loss": 0.037884,
    "eta2d_x": 0.57148,
    "eta2d_y": 0.820616,
    "speed": 1.5e-05,
    "heading_rad": 2.458512,
    "drift_mean": -0.01043122,
    "drift_var": 0.0045301788,
    "opacity_var": 0.0025148056,
    "photosphere_crossed": false
  },
  {
    "step": 2644,
    "epoch": 6,
    "batch_in_epoch": 303,
    "theta1": 0.00445,
    "theta2": 0.008434,
    "loss": 0.022103,
    "eta2d_x": 0.57188,
    "eta2d_y": 0.820337,
    "speed": 4.4e-05,
    "heading_rad": 0.593292,
    "drift_mean": -0.00394508,
    "drift_var": 0.0001499613,
    "opacity_var": 0.0025238221,
    "photosphere_crossed": false
  },
  {
    "step": 2648,
    "epoch": 6,
    "batch_in_epoch": 307,
    "theta1": 0.004608,
    "theta2": 0.008356,
    "loss": 0.030457,
    "eta2d_x": 0.645657,
    "eta2d_y": 0.763628,
    "speed": 7.1e-05,
    "heading_rad": -0.731593,
    "drift_mean": 0.00208849,
    "drift_var": 0.0001780189,
    "opacity_var": 0.000521419,
    "photosphere_crossed": false
  },
  {
    "step": 2652,
    "epoch": 6,
    "batch_in_epoch": 311,
    "theta1": 0.00479,
    "theta2": 0.008198,
    "loss": 0.023886,
    "eta2d_x": 0.7376,
    "eta2d_y": 0.675238,
    "speed": 6.8e-05,
    "heading_rad": -0.831754,
    "drift_mean": -0.00164295,
    "drift_var": 0.0027572717,
    "opacity_var": 0.0013467003,
    "photosphere_crossed": false
  },
  {
    "step": 2656,
    "epoch": 6,
    "batch_in_epoch": 315,
    "theta1": 0.004939,
    "theta2": 0.008028,
    "loss": 0.054516,
    "eta2d_x": 0.812657,
    "eta2d_y": 0.582742,
    "speed": 4.9e-05,
    "heading_rad": -1.158886,
    "drift_mean": 0.00765768,
    "drift_var": 0.0004849675,
    "opacity_var": 0.0014978571,
    "photosphere_crossed": false
  },
  {
    "step": 2660,
    "epoch": 6,
    "batch_in_epoch": 319,
    "theta1": 0.00494,
    "theta2": 0.007938,
    "loss": 0.04499,
    "eta2d_x": 0.83508,
    "eta2d_y": 0.550129,
    "speed": 1.6e-05,
    "heading_rad": -2.666388,
    "drift_mean": -0.00238156,
    "drift_var": 0.0009735448,
    "opacity_var": 0.0008575851,
    "photosphere_crossed": false
  },
  {
    "step": 2664,
    "epoch": 6,
    "batch_in_epoch": 323,
    "theta1": 0.00489,
    "theta2": 0.0079,
    "loss": 0.018222,
    "eta2d_x": 0.836411,
    "eta2d_y": 0.548102,
    "speed": 2.2e-05,
    "heading_rad": -2.610188,
    "drift_mean": -0.00669203,
    "drift_var": 0.0005180638,
    "opacity_var": 0.0007377206,
    "photosphere_crossed": false
  },
  {
    "step": 2668,
    "epoch": 6,
    "batch_in_epoch": 327,
    "theta1": 0.004865,
    "theta2": 0.007923,
    "loss": 0.022256,
    "eta2d_x": 0.826102,
    "eta2d_y": 0.56352,
    "speed": 1.1e-05,
    "heading_rad": 1.733653,
    "drift_mean": 0.00100854,
    "drift_var": 0.0009163925,
    "opacity_var": 0.0007201787,
    "photosphere_crossed": false
  },
  {
    "step": 2672,
    "epoch": 6,
    "batch_in_epoch": 331,
    "theta1": 0.004868,
    "theta2": 0.007953,
    "loss": 0.018656,
    "eta2d_x": 0.818862,
    "eta2d_y": 0.57399,
    "speed": 1e-05,
    "heading_rad": -3.040845,
    "drift_mean": -0.00090013,
    "drift_var": 0.0008745142,
    "opacity_var": 0.0008065593,
    "photosphere_crossed": false
  },
  {
    "step": 2676,
    "epoch": 6,
    "batch_in_epoch": 335,
    "theta1": 0.004794,
    "theta2": 0.007835,
    "loss": 0.090785,
    "eta2d_x": 0.836963,
    "eta2d_y": 0.547259,
    "speed": 4.1e-05,
    "heading_rad": -1.868156,
    "drift_mean": 0.01803228,
    "drift_var": 0.0015333618,
    "opacity_var": 0.0011608627,
    "photosphere_crossed": false
  },
  {
    "step": 2680,
    "epoch": 6,
    "batch_in_epoch": 339,
    "theta1": 0.004701,
    "theta2": 0.007746,
    "loss": 0.014367,
    "eta2d_x": 0.845419,
    "eta2d_y": 0.534104,
    "speed": 3.5e-05,
    "heading_rad": -2.266534,
    "drift_mean": -0.01910436,
    "drift_var": 0.0022790265,
    "opacity_var": 0.0021508487,
    "photosphere_crossed": false
  },
  {
    "step": 2684,
    "epoch": 6,
    "batch_in_epoch": 343,
    "theta1": 0.004583,
    "theta2": 0.007656,
    "loss": 0.016623,
    "eta2d_x": 0.850246,
    "eta2d_y": 0.526385,
    "speed": 4.1e-05,
    "heading_rad": -2.573674,
    "drift_mean": 0.00056394,
    "drift_var": 0.0001776956,
    "opacity_var": 0.0017807162,
    "photosphere_crossed": false
  },
  {
    "step": 2688,
    "epoch": 6,
    "batch_in_epoch": 347,
    "theta1": 0.00431,
    "theta2": 0.007678,
    "loss": 0.029526,
    "eta2d_x": 0.770468,
    "eta2d_y": 0.637479,
    "speed": 7.2e-05,
    "heading_rad": 3.072362,
    "drift_mean": 0.00322586,
    "drift_var": 0.0008447123,
    "opacity_var": 0.0004583505,
    "photosphere_crossed": false
  },
  {
    "step": 2692,
    "epoch": 6,
    "batch_in_epoch": 351,
    "theta1": 0.004085,
    "theta2": 0.007798,
    "loss": 0.054613,
    "eta2d_x": 0.618019,
    "eta2d_y": 0.786163,
    "speed": 5.2e-05,
    "heading_rad": 2.428352,
    "drift_mean": 0.00627172,
    "drift_var": 0.0009212576,
    "opacity_var": 0.0008445525,
    "photosphere_crossed": false
  },
  {
    "step": 2696,
    "epoch": 6,
    "batch_in_epoch": 355,
    "theta1": 0.004013,
    "theta2": 0.007872,
    "loss": 0.013137,
    "eta2d_x": 0.545356,
    "eta2d_y": 0.838204,
    "speed": 1.5e-05,
    "heading_rad": 2.152757,
    "drift_mean": -0.010369,
    "drift_var": 8.37763e-05,
    "opacity_var": 0.0005678488,
    "photosphere_crossed": false
  },
  {
    "step": 2700,
    "epoch": 6,
    "batch_in_epoch": 359,
    "theta1": 0.004018,
    "theta2": 0.007845,
    "loss": 0.037383,
    "eta2d_x": 0.558942,
    "eta2d_y": 0.829207,
    "speed": 2.4e-05,
    "heading_rad": -1.28575,
    "drift_mean": 0.00606148,
    "drift_var": 0.000459564,
    "opacity_var": 0.0003931365,
    "photosphere_crossed": false
  },
  {
    "step": 2704,
    "epoch": 6,
    "batch_in_epoch": 363,
    "theta1": 0.004068,
    "theta2": 0.00777,
    "loss": 0.014822,
    "eta2d_x": 0.621324,
    "eta2d_y": 0.783554,
    "speed": 2.7e-05,
    "heading_rad": -0.775931,
    "drift_mean": -0.00564032,
    "drift_var": 0.0006710887,
    "opacity_var": 0.0005330739,
    "photosphere_crossed": false
  },
  {
    "step": 2708,
    "epoch": 6,
    "batch_in_epoch": 367,
    "theta1": 0.004128,
    "theta2": 0.007798,
    "loss": 0.022926,
    "eta2d_x": 0.640972,
    "eta2d_y": 0.767564,
    "speed": 1.1e-05,
    "heading_rad": 1.266609,
    "drift_mean": 0.002026,
    "drift_var": 0.0005369423,
    "opacity_var": 0.0005785909,
    "photosphere_crossed": false
  },
  {
    "step": 2712,
    "epoch": 6,
    "batch_in_epoch": 371,
    "theta1": 0.004184,
    "theta2": 0.007805,
    "loss": 0.05354,
    "eta2d_x": 0.665674,
    "eta2d_y": 0.746242,
    "speed": 2.3e-05,
    "heading_rad": 0.604005,
    "drift_mean": 0.00765361,
    "drift_var": 0.0025053447,
    "opacity_var": 0.0014419134,
    "photosphere_crossed": false
  },
  {
    "step": 2716,
    "epoch": 6,
    "batch_in_epoch": 375,
    "theta1": 0.004177,
    "theta2": 0.007864,
    "loss": 0.028187,
    "eta2d_x": 0.638405,
    "eta2d_y": 0.769701,
    "speed": 2.4e-05,
    "heading_rad": 2.213019,
    "drift_mean": -0.0063383,
    "drift_var": 0.0013287743,
    "opacity_var": 0.0017521954,
    "photosphere_crossed": false
  },
  {
    "step": 2720,
    "epoch": 6,
    "batch_in_epoch": 379,
    "theta1": 0.004155,
    "theta2": 0.007982,
    "loss": 0.02885,
    "eta2d_x": 0.582469,
    "eta2d_y": 0.812853,
    "speed": 3.6e-05,
    "heading_rad": 1.43202,
    "drift_mean": 0.00016572,
    "drift_var": 0.0007926024,
    "opacity_var": 0.001123329,
    "photosphere_crossed": false
  },
  {
    "step": 2724,
    "epoch": 6,
    "batch_in_epoch": 383,
    "theta1": 0.00423,
    "theta2": 0.008089,
    "loss": 0.020944,
    "eta2d_x": 0.581668,
    "eta2d_y": 0.813426,
    "speed": 3.6e-05,
    "heading_rad": 0.695127,
    "drift_mean": -0.00197652,
    "drift_var": 6.60565e-05,
    "opacity_var": 0.0003842923,
    "photosphere_crossed": false
  },
  {
    "step": 2728,
    "epoch": 6,
    "batch_in_epoch": 387,
    "theta1": 0.004319,
    "theta2": 0.008118,
    "loss": 0.034366,
    "eta2d_x": 0.610964,
    "eta2d_y": 0.791658,
    "speed": 4.1e-05,
    "heading_rad": 0.584942,
    "drift_mean": 0.00335564,
    "drift_var": 0.0005764291,
    "opacity_var": 0.0003078992,
    "photosphere_crossed": false
  },
  {
    "step": 2732,
    "epoch": 6,
    "batch_in_epoch": 391,
    "theta1": 0.004452,
    "theta2": 0.008184,
    "loss": 0.023989,
    "eta2d_x": 0.641844,
    "eta2d_y": 0.766835,
    "speed": 3.2e-05,
    "heading_rad": 0.355035,
    "drift_mean": -0.00259428,
    "drift_var": 0.0002630028,
    "opacity_var": 0.0003824802,
    "photosphere_crossed": false
  },
  {
    "step": 2736,
    "epoch": 6,
    "batch_in_epoch": 395,
    "theta1": 0.004518,
    "theta2": 0.008199,
    "loss": 0.014091,
    "eta2d_x": 0.660069,
    "eta2d_y": 0.751205,
    "speed": 2.5e-05,
    "heading_rad": -0.476927,
    "drift_mean": -0.00247456,
    "drift_var": 0.0005266511,
    "opacity_var": 0.0003517277,
    "photosphere_crossed": false
  },
  {
    "step": 2740,
    "epoch": 6,
    "batch_in_epoch": 399,
    "theta1": 0.004631,
    "theta2": 0.008213,
    "loss": 0.060447,
    "eta2d_x": 0.691181,
    "eta2d_y": 0.722681,
    "speed": 5.7e-05,
    "heading_rad": 0.534025,
    "drift_mean": 0.01158894,
    "drift_var": 0.0001940375,
    "opacity_var": 0.0003843138,
    "photosphere_crossed": false
  },
  {
    "step": 2744,
    "epoch": 6,
    "batch_in_epoch": 403,
    "theta1": 0.004773,
    "theta2": 0.00833,
    "loss": 0.029051,
    "eta2d_x": 0.699076,
    "eta2d_y": 0.715048,
    "speed": 3.6e-05,
    "heading_rad": 0.604849,
    "drift_mean": -0.00784899,
    "drift_var": 0.0003887009,
    "opacity_var": 0.0004988823,
    "photosphere_crossed": false
  },
  {
    "step": 2748,
    "epoch": 6,
    "batch_in_epoch": 407,
    "theta1": 0.004824,
    "theta2": 0.00823,
    "loss": 0.026928,
    "eta2d_x": 0.737317,
    "eta2d_y": 0.675547,
    "speed": 3.5e-05,
    "heading_rad": -2.130595,
    "drift_mean": -0.00053063,
    "drift_var": 0.0033440925,
    "opacity_var": 0.0017151547,
    "photosphere_crossed": false
  },
  {
    "step": 2752,
    "epoch": 6,
    "batch_in_epoch": 411,
    "theta1": 0.004775,
    "theta2": 0.008108,
    "loss": 0.0435,
    "eta2d_x": 0.757952,
    "eta2d_y": 0.65231,
    "speed": 3.8e-05,
    "heading_rad": -2.087838,
    "drift_mean": 0.00414297,
    "drift_var": 0.0002497157,
    "opacity_var": 0.0016184104,
    "photosphere_crossed": false
  },
  {
    "step": 2756,
    "epoch": 6,
    "batch_in_epoch": 415,
    "theta1": 0.00472,
    "theta2": 0.007901,
    "loss": 0.024614,
    "eta2d_x": 0.80384,
    "eta2d_y": 0.594846,
    "speed": 3.8e-05,
    "heading_rad": -1.619505,
    "drift_mean": -0.00472169,
    "drift_var": 0.0002699271,
    "opacity_var": 0.0004765656,
    "photosphere_crossed": false
  },
  {
    "step": 2760,
    "epoch": 6,
    "batch_in_epoch": 419,
    "theta1": 0.004741,
    "theta2": 0.007717,
    "loss": 0.011501,
    "eta2d_x": 0.860642,
    "eta2d_y": 0.50921,
    "speed": 5.2e-05,
    "heading_rad": -1.441834,
    "drift_mean": -0.00327811,
    "drift_var": 0.0010988795,
    "opacity_var": 0.0006788979,
    "photosphere_crossed": false
  },
  {
    "step": 2764,
    "epoch": 6,
    "batch_in_epoch": 423,
    "theta1": 0.004657,
    "theta2": 0.007447,
    "loss": 0.010435,
    "eta2d_x": 0.922812,
    "eta2d_y": 0.38525,
    "speed": 6.3e-05,
    "heading_rad": -2.21305,
    "drift_mean": -0.00026655,
    "drift_var": 0.0006917285,
    "opacity_var": 0.0008112572,
    "photosphere_crossed": false
  },
  {
    "step": 2768,
    "epoch": 6,
    "batch_in_epoch": 427,
    "theta1": 0.004534,
    "theta2": 0.007376,
    "loss": 0.056443,
    "eta2d_x": 0.928463,
    "eta2d_y": 0.371424,
    "speed": 3.3e-05,
    "heading_rad": -2.723071,
    "drift_mean": 0.01150213,
    "drift_var": 0.0003255326,
    "opacity_var": 0.00078472,
    "photosphere_crossed": false
  },
  {
    "step": 2772,
    "epoch": 6,
    "batch_in_epoch": 431,
    "theta1": 0.004414,
    "theta2": 0.007386,
    "loss": 0.03072,
    "eta2d_x": 0.909535,
    "eta2d_y": 0.415627,
    "speed": 3.8e-05,
    "heading_rad": 2.415596,
    "drift_mean": -0.00643094,
    "drift_var": 0.0009273136,
    "opacity_var": 0.0006995346,
    "photosphere_crossed": false
  },
  {
    "step": 2776,
    "epoch": 6,
    "batch_in_epoch": 435,
    "theta1": 0.004422,
    "theta2": 0.007511,
    "loss": 0.046558,
    "eta2d_x": 0.866616,
    "eta2d_y": 0.498976,
    "speed": 3.6e-05,
    "heading_rad": 1.450706,
    "drift_mean": 0.00395969,
    "drift_var": 0.0006514989,
    "opacity_var": 0.0008321695,
    "photosphere_crossed": false
  },
  {
    "step": 2780,
    "epoch": 6,
    "batch_in_epoch": 439,
    "theta1": 0.004475,
    "theta2": 0.007712,
    "loss": 0.049217,
    "eta2d_x": 0.806594,
    "eta2d_y": 0.591106,
    "speed": 6.6e-05,
    "heading_rad": 1.022815,
    "drift_mean": 0.00066474,
    "drift_var": 0.0023140371,
    "opacity_var": 0.0013231816,
    "photosphere_crossed": false
  },
  {
    "step": 2784,
    "epoch": 6,
    "batch_in_epoch": 443,
    "theta1": 0.004522,
    "theta2": 0.007862,
    "loss": 0.013461,
    "eta2d_x": 0.767625,
    "eta2d_y": 0.640899,
    "speed": 3.1e-05,
    "heading_rad": 1.188812,
    "drift_mean": -0.00893906,
    "drift_var": 0.000373402,
    "opacity_var": 0.0013271132,
    "photosphere_crossed": false
  },
  {
    "step": 2788,
    "epoch": 6,
    "batch_in_epoch": 447,
    "theta1": 0.004556,
    "theta2": 0.007915,
    "loss": 0.051425,
    "eta2d_x": 0.759493,
    "eta2d_y": 0.650515,
    "speed": 2e-05,
    "heading_rad": 0.637029,
    "drift_mean": 0.00949093,
    "drift_var": 0.0029093115,
    "opacity_var": 0.0016551688,
    "photosphere_crossed": false
  },
  {
    "step": 2792,
    "epoch": 6,
    "batch_in_epoch": 451,
    "theta1": 0.00464,
    "theta2": 0.007919,
    "loss": 0.023327,
    "eta2d_x": 0.779959,
    "eta2d_y": 0.625831,
    "speed": 3e-05,
    "heading_rad": 0.384384,
    "drift_mean": -0.00702435,
    "drift_var": 0.0002215921,
    "opacity_var": 0.001452616,
    "photosphere_crossed": false
  },
  {
    "step": 2796,
    "epoch": 6,
    "batch_in_epoch": 455,
    "theta1": 0.004694,
    "theta2": 0.008038,
    "loss": 0.080773,
    "eta2d_x": 0.758126,
    "eta2d_y": 0.652108,
    "speed": 3.8e-05,
    "heading_rad": 1.376534,
    "drift_mean": 0.0143615,
    "drift_var": 0.0003966912,
    "opacity_var": 0.000734673,
    "photosphere_crossed": false
  },
  {
    "step": 2800,
    "epoch": 6,
    "batch_in_epoch": 459,
    "theta1": 0.004741,
    "theta2": 0.008191,
    "loss": 0.066344,
    "eta2d_x": 0.727142,
    "eta2d_y": 0.686488,
    "speed": 3.8e-05,
    "heading_rad": 1.507199,
    "drift_mean": -0.00360729,
    "drift_var": 0.0019211347,
    "opacity_var": 0.0011110451,
    "photosphere_crossed": false
  },
  {
    "step": 2804,
    "epoch": 6,
    "batch_in_epoch": 463,
    "theta1": 0.00478,
    "theta2": 0.008366,
    "loss": 0.03877,
    "eta2d_x": 0.692026,
    "eta2d_y": 0.721872,
    "speed": 4.4e-05,
    "heading_rad": 1.346698,
    "drift_mean": -0.00689364,
    "drift_var": 0.0009545011,
    "opacity_var": 0.0014484376,
    "photosphere_crossed": false
  },
  {
    "step": 2808,
    "epoch": 6,
    "batch_in_epoch": 467,
    "theta1": 0.004711,
    "theta2": 0.008468,
    "loss": 0.061511,
    "eta2d_x": 0.64807,
    "eta2d_y": 0.761581,
    "speed": 5.4e-05,
    "heading_rad": 2.047308,
    "drift_mean": 0.00568527,
    "drift_var": 0.0014969767,
    "opacity_var": 0.0011347081,
    "photosphere_crossed": false
  },
  {
    "step": 2812,
    "epoch": 7,
    "batch_in_epoch": 3,
    "theta1": 0.004649,
    "theta2": 0.008656,
    "loss": 0.020773,
    "eta2d_x": 0.585847,
    "eta2d_y": 0.810422,
    "speed": 5.3e-05,
    "heading_rad": 1.925893,
    "drift_mean": -0.01018448,
    "drift_var": 0.0005483919,
    "opacity_var": 0.0010297154,
    "photosphere_crossed": false
  },
  {
    "step": 2816,
    "epoch": 7,
    "batch_in_epoch": 7,
    "theta1": 0.004485,
    "theta2": 0.008839,
    "loss": 0.012857,
    "eta2d_x": 0.49397,
    "eta2d_y": 0.869479,
    "speed": 5.9e-05,
    "heading_rad": 2.341342,
    "drift_mean": -0.00197891,
    "drift_var": 0.0009804398,
    "opacity_var": 0.0007209928,
    "photosphere_crossed": false
  },
  {
    "step": 2820,
    "epoch": 7,
    "batch_in_epoch": 11,
    "theta1": 0.004356,
    "theta2": 0.008936,
    "loss": 0.035203,
    "eta2d_x": 0.430542,
    "eta2d_y": 0.902571,
    "speed": 3.6e-05,
    "heading_rad": 2.995013,
    "drift_mean": 0.00558644,
    "drift_var": 0.0002335565,
    "opacity_var": 0.0005530104,
    "photosphere_crossed": false
  },
  {
    "step": 2824,
    "epoch": 7,
    "batch_in_epoch": 15,
    "theta1": 0.004332,
    "theta2": 0.008879,
    "loss": 0.018526,
    "eta2d_x": 0.431621,
    "eta2d_y": 0.902055,
    "speed": 2.1e-05,
    "heading_rad": -1.638034,
    "drift_mean": -0.00416931,
    "drift_var": 5.76235e-05,
    "opacity_var": 0.0003911243,
    "photosphere_crossed": false
  },
  {
    "step": 2828,
    "epoch": 7,
    "batch_in_epoch": 19,
    "theta1": 0.00432,
    "theta2": 0.008769,
    "loss": 0.046641,
    "eta2d_x": 0.447681,
    "eta2d_y": 0.894193,
    "speed": 3.9e-05,
    "heading_rad": -0.950399,
    "drift_mean": 0.00702879,
    "drift_var": 0.0004990694,
    "opacity_var": 0.0003107265,
    "photosphere_crossed": false
  },
  {
    "step": 2832,
    "epoch": 7,
    "batch_in_epoch": 23,
    "theta1": 0.004494,
    "theta2": 0.008622,
    "loss": 0.011227,
    "eta2d_x": 0.542676,
    "eta2d_y": 0.839942,
    "speed": 5e-05,
    "heading_rad": -0.446796,
    "drift_mean": -0.0088534,
    "drift_var": 0.0017881423,
    "opacity_var": 0.0010729922,
    "photosphere_crossed": false
  },
  {
    "step": 2836,
    "epoch": 7,
    "batch_in_epoch": 27,
    "theta1": 0.004704,
    "theta2": 0.008562,
    "loss": 0.026817,
    "eta2d_x": 0.623543,
    "eta2d_y": 0.781789,
    "speed": 5.8e-05,
    "heading_rad": -0.362356,
    "drift_mean": 0.00389741,
    "drift_var": 4.42179e-05,
    "opacity_var": 0.0009148464,
    "photosphere_crossed": false
  },
  {
    "step": 2840,
    "epoch": 7,
    "batch_in_epoch": 31,
    "theta1": 0.004866,
    "theta2": 0.008569,
    "loss": 0.016174,
    "eta2d_x": 0.665719,
    "eta2d_y": 0.746203,
    "speed": 2.8e-05,
    "heading_rad": 0.601606,
    "drift_mean": -0.00266078,
    "drift_var": 0.0002894868,
    "opacity_var": 0.0003048973,
    "photosphere_crossed": false
  },
  {
    "step": 2844,
    "epoch": 7,
    "batch_in_epoch": 35,
    "theta1": 0.00491,
    "theta2": 0.008598,
    "loss": 0.014083,
    "eta2d_x": 0.670274,
    "eta2d_y": 0.742114,
    "speed": 1.6e-05,
    "heading_rad": 1.083728,
    "drift_mean": -0.00052265,
    "drift_var": 0.000120081,
    "opacity_var": 0.0001837957,
    "photosphere_crossed": true
  },
  {
    "step": 2848,
    "epoch": 7,
    "batch_in_epoch": 39,
    "theta1": 0.004872,
    "theta2": 0.008561,
    "loss": 0.026323,
    "eta2d_x": 0.66924,
    "eta2d_y": 0.743047,
    "speed": 2.6e-05,
    "heading_rad": -2.553561,
    "drift_mean": 0.00305983,
    "drift_var": 0.0001333023,
    "opacity_var": 0.0001596047,
    "photosphere_crossed": true
  },
  {
    "step": 2852,
    "epoch": 7,
    "batch_in_epoch": 43,
    "theta1": 0.00484,
    "theta2": 0.008518,
    "loss": 0.052497,
    "eta2d_x": 0.670908,
    "eta2d_y": 0.74154,
    "speed": 1.1e-05,
    "heading_rad": -1.793348,
    "drift_mean": 0.00654359,
    "drift_var": 0.000266064,
    "opacity_var": 0.0001859304,
    "photosphere_crossed": true
  },
  {
    "step": 2856,
    "epoch": 7,
    "batch_in_epoch": 47,
    "theta1": 0.004805,
    "theta2": 0.008502,
    "loss": 0.022186,
    "eta2d_x": 0.665648,
    "eta2d_y": 0.746266,
    "speed": 2.3e-05,
    "heading_rad": -2.965112,
    "drift_mean": -0.00757763,
    "drift_var": 0.0005004586,
    "opacity_var": 0.0003879796,
    "photosphere_crossed": false
  },
  {
    "step": 2860,
    "epoch": 7,
    "batch_in_epoch": 51,
    "theta1": 0.004697,
    "theta2": 0.008536,
    "loss": 0.038641,
    "eta2d_x": 0.627573,
    "eta2d_y": 0.778558,
    "speed": 3e-05,
    "heading_rad": 2.217908,
    "drift_mean": 0.00411371,
    "drift_var": 0.000226701,
    "opacity_var": 0.0003673287,
    "photosphere_crossed": false
  },
  {
    "step": 2864,
    "epoch": 7,
    "batch_in_epoch": 55,
    "theta1": 0.004717,
    "theta2": 0.008614,
    "loss": 0.014764,
    "eta2d_x": 0.61554,
    "eta2d_y": 0.788106,
    "speed": 1.9e-05,
    "heading_rad": 1.75733,
    "drift_mean": -0.0059692,
    "drift_var": 0.000712329,
    "opacity_var": 0.0004559221,
    "photosphere_crossed": false
  },
  {
    "step": 2868,
    "epoch": 7,
    "batch_in_epoch": 59,
    "theta1": 0.004773,
    "theta2": 0.008734,
    "loss": 0.044157,
    "eta2d_x": 0.605178,
    "eta2d_y": 0.79609,
    "speed": 4e-05,
    "heading_rad": 0.947532,
    "drift_mean": 0.00734818,
    "drift_var": 3.99374e-05,
    "opacity_var": 0.0003744847,
    "photosphere_crossed": false
  },
  {
    "step": 2872,
    "epoch": 7,
    "batch_in_epoch": 63,
    "theta1": 0.004874,
    "theta2": 0.00885,
    "loss": 0.007992,
    "eta2d_x": 0.608394,
    "eta2d_y": 0.793635,
    "speed": 3.7e-05,
    "heading_rad": 0.840261,
    "drift_mean": -0.00904139,
    "drift_var": 9.76501e-05,
    "opacity_var": 0.0001372299,
    "photosphere_crossed": true
  },
  {
    "step": 2876,
    "epoch": 7,
    "batch_in_epoch": 67,
    "theta1": 0.004941,
    "theta2": 0.008896,
    "loss": 0.005606,
    "eta2d_x": 0.616436,
    "eta2d_y": 0.787405,
    "speed": 1.8e-05,
    "heading_rad": 0.201965,
    "drift_mean": -0.00059632,
    "drift_var": 0.0002008068,
    "opacity_var": 0.0001653506,
    "photosphere_crossed": true
  },
  {
    "step": 2880,
    "epoch": 7,
    "batch_in_epoch": 71,
    "theta1": 0.005024,
    "theta2": 0.008886,
    "loss": 0.028513,
    "eta2d_x": 0.638296,
    "eta2d_y": 0.769791,
    "speed": 1.7e-05,
    "heading_rad": -0.291319,
    "drift_mean": 0.00572657,
    "drift_var": 4.58002e-05,
    "opacity_var": 0.0001187097,
    "photosphere_crossed": true
  },
  {
    "step": 2884,
    "epoch": 7,
    "batch_in_epoch": 75,
    "theta1": 0.005163,
    "theta2": 0.008837,
    "loss": 0.020929,
    "eta2d_x": 0.678122,
    "eta2d_y": 0.734949,
    "speed": 3.5e-05,
    "heading_rad": -0.274741,
    "drift_mean": -0.001896,
    "drift_var": 0.0005138412,
    "opacity_var": 0.0003273652,
    "photosphere_crossed": false
  },
  {
    "step": 2888,
    "epoch": 7,
    "batch_in_epoch": 79,
    "theta1": 0.005214,
    "theta2": 0.008782,
    "loss": 0.011197,
    "eta2d_x": 0.69894,
    "eta2d_y": 0.71518,
    "speed": 1.9e-05,
    "heading_rad": -1.05681,
    "drift_mean": -0.00243281,
    "drift_var": 0.0005285448,
    "opacity_var": 0.0004977962,
    "photosphere_crossed": false
  },
  {
    "step": 2892,
    "epoch": 7,
    "batch_in_epoch": 83,
    "theta1": 0.005247,
    "theta2": 0.008628,
    "loss": 0.009106,
    "eta2d_x": 0.735394,
    "eta2d_y": 0.677639,
    "speed": 4e-05,
    "heading_rad": -1.387232,
    "drift_mean": -0.00052296,
    "drift_var": 6.04488e-05,
    "opacity_var": 0.0002749757,
    "photosphere_crossed": true
  },
  {
    "step": 2896,
    "epoch": 7,
    "batch_in_epoch": 87,
    "theta1": 0.005235,
    "theta2": 0.008424,
    "loss": 0.005693,
    "eta2d_x": 0.77435,
    "eta2d_y": 0.632758,
    "speed": 5.1e-05,
    "heading_rad": -1.661678,
    "drift_mean": -0.00085309,
    "drift_var": 0.0002456085,
    "opacity_var": 0.0001416345,
    "photosphere_crossed": true
  },
  {
    "step": 2900,
    "epoch": 7,
    "batch_in_epoch": 91,
    "theta1": 0.005204,
    "theta2": 0.008252,
    "loss": 0.021796,
    "eta2d_x": 0.80526,
    "eta2d_y": 0.592922,
    "speed": 5.3e-05,
    "heading_rad": -1.890008,
    "drift_mean": 0.00402569,
    "drift_var": 6.91645e-05,
    "opacity_var": 0.0001458864,
    "photosphere_crossed": true
  },
  {
    "step": 2904,
    "epoch": 7,
    "batch_in_epoch": 95,
    "theta1": 0.005108,
    "theta2": 0.00815,
    "loss": 0.024138,
    "eta2d_x": 0.812454,
    "eta2d_y": 0.583026,
    "speed": 2.2e-05,
    "heading_rad": 3.069972,
    "drift_mean": 0.00058542,
    "drift_var": 8.48053e-05,
    "opacity_var": 0.0001489505,
    "photosphere_crossed": true
  },
  {
    "step": 2908,
    "epoch": 7,
    "batch_in_epoch": 99,
    "theta1": 0.005,
    "theta2": 0.008152,
    "loss": 0.02221,
    "eta2d_x": 0.793321,
    "eta2d_y": 0.608804,
    "speed": 2.1e-05,
    "heading_rad": -3.059707,
    "drift_mean": -0.00048183,
    "drift_var": 0.0012275481,
    "opacity_var": 0.0005899877,
    "photosphere_crossed": false
  },
  {
    "step": 2912,
    "epoch": 7,
    "batch_in_epoch": 103,
    "theta1": 0.005078,
    "theta2": 0.008161,
    "loss": 0.015088,
    "eta2d_x": 0.804925,
    "eta2d_y": 0.593376,
    "speed": 3.6e-05,
    "heading_rad": 0.085698,
    "drift_mean": -0.00178045,
    "drift_var": 0.0004967626,
    "opacity_var": 0.0007895532,
    "photosphere_crossed": false
  },
  {
    "step": 2916,
    "epoch": 7,
    "batch_in_epoch": 107,
    "theta1": 0.005361,
    "theta2": 0.008325,
    "loss": 0.019285,
    "eta2d_x": 0.813697,
    "eta2d_y": 0.58129,
    "speed": 9.3e-05,
    "heading_rad": 0.673752,
    "drift_mean": 0.00104915,
    "drift_var": 0.0091606241,
    "opacity_var": 0.0043237369,
    "photosphere_crossed": false
  },
  {
    "step": 2920,
    "epoch": 7,
    "batch_in_epoch": 111,
    "theta1": 0.005568,
    "theta2": 0.008548,
    "loss": 0.034689,
    "eta2d_x": 0.800401,
    "eta2d_y": 0.599465,
    "speed": 8.2e-05,
    "heading_rad": 1.086294,
    "drift_mean": 0.003851,
    "drift_var": 0.0018349271,
    "opacity_var": 0.0049799693,
    "photosphere_crossed": false
  },
  {
    "step": 2924,
    "epoch": 7,
    "batch_in_epoch": 115,
    "theta1": 0.005645,
    "theta2": 0.008711,
    "loss": 0.010364,
    "eta2d_x": 0.782452,
    "eta2d_y": 0.622711,
    "speed": 3.4e-05,
    "heading_rad": 1.210956,
    "drift_mean": -0.00608125,
    "drift_var": 0.0017066665,
    "opacity_var": 0.0015996526,
    "photosphere_crossed": false
  },
  {
    "step": 2928,
    "epoch": 7,
    "batch_in_epoch": 119,
    "theta1": 0.00566,
    "theta2": 0.008803,
    "loss": 0.044071,
    "eta2d_x": 0.76909,
    "eta2d_y": 0.63914,
    "speed": 1e-05,
    "heading_rad": 2.768783,
    "drift_mean": 0.00842673,
    "drift_var": 0.0003978198,
    "opacity_var": 0.0012283323,
    "photosphere_crossed": false
  },
  {
    "step": 2932,
    "epoch": 7,
    "batch_in_epoch": 123,
    "theta1": 0.005579,
    "theta2": 0.008787,
    "loss": 0.036554,
    "eta2d_x": 0.760334,
    "eta2d_y": 0.649533,
    "speed": 2.5e-05,
    "heading_rad": -2.829102,
    "drift_mean": -0.00187916,
    "drift_var": 0.0019554921,
    "opacity_var": 0.0014103041,
    "photosphere_crossed": false
  },
  {
    "step": 2936,
    "epoch": 7,
    "batch_in_epoch": 127,
    "theta1": 0.00558,
    "theta2": 0.008747,
    "loss": 0.013289,
    "eta2d_x": 0.767169,
    "eta2d_y": 0.641445,
    "speed": 9e-06,
    "heading_rad": -1.355527,
    "drift_mean": -0.00581629,
    "drift_var": 0.0002108717,
    "opacity_var": 0.0010001972,
    "photosphere_crossed": false
  },
  {
    "step": 2940,
    "epoch": 7,
    "batch_in_epoch": 131,
    "theta1": 0.00554,
    "theta2": 0.008758,
    "loss": 0.014745,
    "eta2d_x": 0.759428,
    "eta2d_y": 0.650591,
    "speed": 2e-05,
    "heading_rad": 3.034283,
    "drift_mean": 0.00036406,
    "drift_var": 0.0031346174,
    "opacity_var": 0.0017019193,
    "photosphere_crossed": false
  },
  {
    "step": 2944,
    "epoch": 7,
    "batch_in_epoch": 135,
    "theta1": 0.005464,
    "theta2": 0.008776,
    "loss": 0.017131,
    "eta2d_x": 0.744569,
    "eta2d_y": 0.667545,
    "speed": 2e-05,
    "heading_rad": 2.976584,
    "drift_mean": 0.00059634,
    "drift_var": 0.0010270189,
    "opacity_var": 0.0019051168,
    "photosphere_crossed": false
  },
  {
    "step": 2948,
    "epoch": 7,
    "batch_in_epoch": 139,
    "theta1": 0.005372,
    "theta2": 0.008791,
    "loss": 0.008207,
    "eta2d_x": 0.726399,
    "eta2d_y": 0.687273,
    "speed": 2.7e-05,
    "heading_rad": 2.909949,
    "drift_mean": -0.00223103,
    "drift_var": 0.0003305347,
    "opacity_var": 0.0011596593,
    "photosphere_crossed": false
  },
  {
    "step": 2952,
    "epoch": 7,
    "batch_in_epoch": 143,
    "theta1": 0.005269,
    "theta2": 0.00895,
    "loss": 0.049327,
    "eta2d_x": 0.678482,
    "eta2d_y": 0.734617,
    "speed": 6.7e-05,
    "heading_rad": 2.001389,
    "drift_mean": 0.01028004,
    "drift_var": 0.0005470373,
    "opacity_var": 0.0005995907,
    "photosphere_crossed": false
  },
  {
    "step": 2956,
    "epoch": 7,
    "batch_in_epoch": 147,
    "theta1": 0.005209,
    "theta2": 0.009118,
    "loss": 0.010458,
    "eta2d_x": 0.636731,
    "eta2d_y": 0.771086,
    "speed": 3.6e-05,
    "heading_rad": 1.618572,
    "drift_mean": -0.00971714,
    "drift_var": 0.0034771022,
    "opacity_var": 0.0019596418,
    "photosphere_crossed": false
  },
  {
    "step": 2960,
    "epoch": 7,
    "batch_in_epoch": 151,
    "theta1": 0.005278,
    "theta2": 0.009186,
    "loss": 0.009326,
    "eta2d_x": 0.639336,
    "eta2d_y": 0.768928,
    "speed": 1.5e-05,
    "heading_rad": 0.489557,
    "drift_mean": -0.00028297,
    "drift_var": 0.0006969612,
    "opacity_var": 0.0020327135,
    "photosphere_crossed": false
  },
  {
    "step": 2964,
    "epoch": 7,
    "batch_in_epoch": 155,
    "theta1": 0.005274,
    "theta2": 0.009159,
    "loss": 0.017618,
    "eta2d_x": 0.642944,
    "eta2d_y": 0.765914,
    "speed": 1.4e-05,
    "heading_rad": -2.678348,
    "drift_mean": 0.00207294,
    "drift_var": 0.0005653995,
    "opacity_var": 0.0011721912,
    "photosphere_crossed": false
  },
  {
    "step": 2968,
    "epoch": 7,
    "batch_in_epoch": 159,
    "theta1": 0.005254,
    "theta2": 0.009139,
    "loss": 0.012716,
    "eta2d_x": 0.642504,
    "eta2d_y": 0.766283,
    "speed": 7e-06,
    "heading_rad": -2.65706,
    "drift_mean": -0.00122566,
    "drift_var": 0.000119986,
    "opacity_var": 0.0004391167,
    "photosphere_crossed": false
  },
  {
    "step": 2972,
    "epoch": 7,
    "batch_in_epoch": 163,
    "theta1": 0.005246,
    "theta2": 0.009115,
    "loss": 0.02127,
    "eta2d_x": 0.644894,
    "eta2d_y": 0.764272,
    "speed": 1.2e-05,
    "heading_rad": -2.705157,
    "drift_mean": 0.00213865,
    "drift_var": 8.13302e-05,
    "opacity_var": 9.73787e-05,
    "photosphere_crossed": true
  },
  {
    "step": 2976,
    "epoch": 7,
    "batch_in_epoch": 167,
    "theta1": 0.005139,
    "theta2": 0.009133,
    "loss": 0.023911,
    "eta2d_x": 0.61914,
    "eta2d_y": 0.785281,
    "speed": 5e-05,
    "heading_rad": 3.086491,
    "drift_mean": 0.00066025,
    "drift_var": 0.0001466412,
    "opacity_var": 0.0001141507,
    "photosphere_crossed": true
  },
  {
    "step": 2980,
    "epoch": 7,
    "batch_in_epoch": 171,
    "theta1": 0.005028,
    "theta2": 0.009169,
    "loss": 0.011968,
    "eta2d_x": 0.587673,
    "eta2d_y": 0.809099,
    "speed": 2.3e-05,
    "heading_rad": 2.764907,
    "drift_mean": -0.00298589,
    "drift_var": 0.000568613,
    "opacity_var": 0.000321389,
    "photosphere_crossed": false
  },
  {
    "step": 2984,
    "epoch": 7,
    "batch_in_epoch": 175,
    "theta1": 0.005013,
    "theta2": 0.009216,
    "loss": 0.05036,
    "eta2d_x": 0.57623,
    "eta2d_y": 0.817287,
    "speed": 5e-06,
    "heading_rad": -2.865059,
    "drift_mean": 0.0095982,
    "drift_var": 0.00070512,
    "opacity_var": 0.0006072068,
    "photosphere_crossed": false
  },
  {
    "step": 2988,
    "epoch": 7,
    "batch_in_epoch": 179,
    "theta1": 0.004974,
    "theta2": 0.009163,
    "loss": 0.018541,
    "eta2d_x": 0.575862,
    "eta2d_y": 0.817547,
    "speed": 4.2e-05,
    "heading_rad": -2.060921,
    "drift_mean": -0.00795496,
    "drift_var": 0.0032020539,
    "opacity_var": 0.0018791242,
    "photosphere_crossed": false
  },
  {
    "step": 2992,
    "epoch": 7,
    "batch_in_epoch": 183,
    "theta1": 0.004886,
    "theta2": 0.009066,
    "loss": 0.015067,
    "eta2d_x": 0.570752,
    "eta2d_y": 0.821122,
    "speed": 2.1e-05,
    "heading_rad": -2.519987,
    "drift_mean": -0.00086838,
    "drift_var": 0.0002281361,
    "opacity_var": 0.0015511433,
    "photosphere_crossed": false
  },
  {
    "step": 2996,
    "epoch": 7,
    "batch_in_epoch": 187,
    "theta1": 0.004821,
    "theta2": 0.008971,
    "loss": 0.00708,
    "eta2d_x": 0.571196,
    "eta2d_y": 0.820814,
    "speed": 3.6e-05,
    "heading_rad": -1.895531,
    "drift_mean": -0.00199671,
    "drift_var": 0.0002741928,
    "opacity_var": 0.0006674287,
    "photosphere_crossed": false
  },
  {
    "step": 3000,
    "epoch": 7,
    "batch_in_epoch": 191,
    "theta1": 0.004798,
    "theta2": 0.008914,
    "loss": 0.024072,
    "eta2d_x": 0.575742,
    "eta2d_y": 0.817632,
    "speed": 1.2e-05,
    "heading_rad": -1.245467,
    "drift_mean": 0.00424806,
    "drift_var": 0.0005511432,
    "opacity_var": 0.0003802186,
    "photosphere_crossed": false
  },
  {
    "step": 3004,
    "epoch": 7,
    "batch_in_epoch": 195,
    "theta1": 0.004822,
    "theta2": 0.008854,
    "loss": 0.045966,
    "eta2d_x": 0.593827,
    "eta2d_y": 0.804593,
    "speed": 3e-06,
    "heading_rad": 0.652942,
    "drift_mean": 0.00547345,
    "drift_var": 0.0003863501,
    "opacity_var": 0.0005069058,
    "photosphere_crossed": false
  },
  {
    "step": 3008,
    "epoch": 7,
    "batch_in_epoch": 199,
    "theta1": 0.004745,
    "theta2": 0.00882,
    "loss": 0.030289,
    "eta2d_x": 0.579413,
    "eta2d_y": 0.815034,
    "speed": 4.7e-05,
    "heading_rad": -3.0843,
    "drift_mean": -0.0039192,
    "drift_var": 0.0003367147,
    "opacity_var": 0.0003420656,
    "photosphere_crossed": false
  },
  {
    "step": 3012,
    "epoch": 7,
    "batch_in_epoch": 203,
    "theta1": 0.004621,
    "theta2": 0.008824,
    "loss": 0.044889,
    "eta2d_x": 0.541721,
    "eta2d_y": 0.840558,
    "speed": 3.1e-05,
    "heading_rad": -2.66018,
    "drift_mean": 0.00364981,
    "drift_var": 0.0006161484,
    "opacity_var": 0.000575792,
    "photosphere_crossed": false
  },
  {
    "step": 3016,
    "epoch": 7,
    "batch_in_epoch": 207,
    "theta1": 0.00454,
    "theta2": 0.008734,
    "loss": 0.019427,
    "eta2d_x": 0.533981,
    "eta2d_y": 0.845497,
    "speed": 2.6e-05,
    "heading_rad": -1.664847,
    "drift_mean": -0.00636539,
    "drift_var": 0.0023007393,
    "opacity_var": 0.0013188447,
    "photosphere_crossed": false
  },
  {
    "step": 3020,
    "epoch": 7,
    "batch_in_epoch": 211,
    "theta1": 0.004536,
    "theta2": 0.008663,
    "loss": 0.032355,
    "eta2d_x": 0.548003,
    "eta2d_y": 0.836477,
    "speed": 4e-06,
    "heading_rad": 0.100236,
    "drift_mean": 0.00323186,
    "drift_var": 4.44327e-05,
    "opacity_var": 0.0012275093,
    "photosphere_crossed": false
  },
  {
    "step": 3024,
    "epoch": 7,
    "batch_in_epoch": 215,
    "theta1": 0.00454,
    "theta2": 0.00866,
    "loss": 0.014083,
    "eta2d_x": 0.550039,
    "eta2d_y": 0.835139,
    "speed": 1e-05,
    "heading_rad": -1.548379,
    "drift_mean": -0.00456785,
    "drift_var": 8.95417e-05,
    "opacity_var": 0.0004174463,
    "photosphere_crossed": false
  },
  {
    "step": 3028,
    "epoch": 7,
    "batch_in_epoch": 219,
    "theta1": 0.0045,
    "theta2": 0.008639,
    "loss": 0.061471,
    "eta2d_x": 0.54112,
    "eta2d_y": 0.840945,
    "speed": 2.1e-05,
    "heading_rad": -2.604715,
    "drift_mean": 0.01184688,
    "drift_var": 0.0004984904,
    "opacity_var": 0.000321455,
    "photosphere_crossed": false
  },
  {
    "step": 3032,
    "epoch": 7,
    "batch_in_epoch": 223,
    "theta1": 0.004388,
    "theta2": 0.008572,
    "loss": 0.011854,
    "eta2d_x": 0.515502,
    "eta2d_y": 0.856888,
    "speed": 5e-05,
    "heading_rad": -2.678402,
    "drift_mean": -0.01240418,
    "drift_var": 0.0007987563,
    "opacity_var": 0.0007124121,
    "photosphere_crossed": false
  },
  {
    "step": 3036,
    "epoch": 7,
    "batch_in_epoch": 227,
    "theta1": 0.004265,
    "theta2": 0.008559,
    "loss": 0.021094,
    "eta2d_x": 0.469298,
    "eta2d_y": 0.88304,
    "speed": 3.7e-05,
    "heading_rad": -3.109869,
    "drift_mean": 0.00231002,
    "drift_var": 0.0001922358,
    "opacity_var": 0.0007858084,
    "photosphere_crossed": false
  },
  {
    "step": 3040,
    "epoch": 7,
    "batch_in_epoch": 231,
    "theta1": 0.00417,
    "theta2": 0.00851,
    "loss": 0.013813,
    "eta2d_x": 0.438227,
    "eta2d_y": 0.898864,
    "speed": 1.9e-05,
    "heading_rad": -1.973171,
    "drift_mean": -0.00182025,
    "drift_var": 1.05844e-05,
    "opacity_var": 0.0001785201,
    "photosphere_crossed": true
  },
  {
    "step": 3044,
    "epoch": 7,
    "batch_in_epoch": 235,
    "theta1": 0.004195,
    "theta2": 0.008506,
    "loss": 0.019252,
    "eta2d_x": 0.450558,
    "eta2d_y": 0.892747,
    "speed": 1.8e-05,
    "heading_rad": 0.122589,
    "drift_mean": 0.00135971,
    "drift_var": 1.94528e-05,
    "opacity_var": 2.22844e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3048,
    "epoch": 7,
    "batch_in_epoch": 239,
    "theta1": 0.004179,
    "theta2": 0.008536,
    "loss": 0.017719,
    "eta2d_x": 0.436975,
    "eta2d_y": 0.899474,
    "speed": 1.3e-05,
    "heading_rad": 0.536013,
    "drift_mean": -0.00038323,
    "drift_var": 0.0005118058,
    "opacity_var": 0.0002376394,
    "photosphere_crossed": true
  },
  {
    "step": 3052,
    "epoch": 7,
    "batch_in_epoch": 243,
    "theta1": 0.004236,
    "theta2": 0.00855,
    "loss": 0.009321,
    "eta2d_x": 0.458656,
    "eta2d_y": 0.888614,
    "speed": 1.1e-05,
    "heading_rad": -0.042814,
    "drift_mean": -0.00209948,
    "drift_var": 5.4455e-05,
    "opacity_var": 0.0002524676,
    "photosphere_crossed": true
  },
  {
    "step": 3056,
    "epoch": 7,
    "batch_in_epoch": 247,
    "theta1": 0.004178,
    "theta2": 0.008463,
    "loss": 0.046932,
    "eta2d_x": 0.452882,
    "eta2d_y": 0.89157,
    "speed": 1.8e-05,
    "heading_rad": -2.283562,
    "drift_mean": 0.00940263,
    "drift_var": 0.0004726457,
    "opacity_var": 0.0002637823,
    "photosphere_crossed": true
  },
  {
    "step": 3060,
    "epoch": 7,
    "batch_in_epoch": 251,
    "theta1": 0.004184,
    "theta2": 0.008421,
    "loss": 0.025926,
    "eta2d_x": 0.46563,
    "eta2d_y": 0.884979,
    "speed": 1.2e-05,
    "heading_rad": -2.463866,
    "drift_mean": -0.00525131,
    "drift_var": 0.0007053956,
    "opacity_var": 0.0005886721,
    "photosphere_crossed": false
  },
  {
    "step": 3064,
    "epoch": 7,
    "batch_in_epoch": 255,
    "theta1": 0.004175,
    "theta2": 0.008427,
    "loss": 0.019895,
    "eta2d_x": 0.460254,
    "eta2d_y": 0.887787,
    "speed": 7e-06,
    "heading_rad": 1.511965,
    "drift_mean": -0.00150773,
    "drift_var": 0.000138484,
    "opacity_var": 0.000539386,
    "photosphere_crossed": false
  },
  {
    "step": 3068,
    "epoch": 7,
    "batch_in_epoch": 259,
    "theta1": 0.004127,
    "theta2": 0.008459,
    "loss": 0.024873,
    "eta2d_x": 0.429783,
    "eta2d_y": 0.902932,
    "speed": 1.8e-05,
    "heading_rad": 2.56639,
    "drift_mean": 0.0012444,
    "drift_var": 0.0003293806,
    "opacity_var": 0.0002422674,
    "photosphere_crossed": true
  },
  {
    "step": 3072,
    "epoch": 7,
    "batch_in_epoch": 263,
    "theta1": 0.004065,
    "theta2": 0.008502,
    "loss": 0.022429,
    "eta2d_x": 0.390489,
    "eta2d_y": 0.920607,
    "speed": 1.3e-05,
    "heading_rad": -2.304798,
    "drift_mean": -0.00061091,
    "drift_var": 9.67106e-05,
    "opacity_var": 0.0001946543,
    "photosphere_crossed": true
  },
  {
    "step": 3076,
    "epoch": 7,
    "batch_in_epoch": 267,
    "theta1": 0.003983,
    "theta2": 0.008405,
    "loss": 0.011599,
    "eta2d_x": 0.368716,
    "eta2d_y": 0.929542,
    "speed": 3.3e-05,
    "heading_rad": -2.496436,
    "drift_mean": -0.00270752,
    "drift_var": 0.0002547249,
    "opacity_var": 0.0001965409,
    "photosphere_crossed": true
  },
  {
    "step": 3080,
    "epoch": 7,
    "batch_in_epoch": 271,
    "theta1": 0.003897,
    "theta2": 0.00835,
    "loss": 0.014995,
    "eta2d_x": 0.33157,
    "eta2d_y": 0.943431,
    "speed": 3.6e-05,
    "heading_rad": -2.57795,
    "drift_mean": 0.00084903,
    "drift_var": 5.45756e-05,
    "opacity_var": 0.0001601525,
    "photosphere_crossed": true
  },
  {
    "step": 3084,
    "epoch": 7,
    "batch_in_epoch": 275,
    "theta1": 0.003818,
    "theta2": 0.008405,
    "loss": 0.028552,
    "eta2d_x": 0.27436,
    "eta2d_y": 0.961627,
    "speed": 3.3e-05,
    "heading_rad": 2.076531,
    "drift_mean": 0.00338926,
    "drift_var": 0.0001656046,
    "opacity_var": 0.0001684695,
    "photosphere_crossed": true
  },
  {
    "step": 3088,
    "epoch": 7,
    "batch_in_epoch": 279,
    "theta1": 0.003836,
    "theta2": 0.008481,
    "loss": 0.005799,
    "eta2d_x": 0.272333,
    "eta2d_y": 0.962203,
    "speed": 2.2e-05,
    "heading_rad": 0.9977,
    "drift_mean": -0.00568839,
    "drift_var": 0.0003190952,
    "opacity_var": 0.000235153,
    "photosphere_crossed": true
  },
  {
    "step": 3092,
    "epoch": 7,
    "batch_in_epoch": 283,
    "theta1": 0.003892,
    "theta2": 0.008572,
    "loss": 0.048467,
    "eta2d_x": 0.288793,
    "eta2d_y": 0.957391,
    "speed": 3.9e-05,
    "heading_rad": 1.15563,
    "drift_mean": 0.01066712,
    "drift_var": 0.0001147778,
    "opacity_var": 0.0002628133,
    "photosphere_crossed": true
  },
  {
    "step": 3096,
    "epoch": 7,
    "batch_in_epoch": 287,
    "theta1": 0.003979,
    "theta2": 0.008719,
    "loss": 0.014605,
    "eta2d_x": 0.30897,
    "eta2d_y": 0.951072,
    "speed": 2.7e-05,
    "heading_rad": 0.848723,
    "drift_mean": -0.00846563,
    "drift_var": 0.0004314029,
    "opacity_var": 0.0003243574,
    "photosphere_crossed": false
  },
  {
    "step": 3100,
    "epoch": 7,
    "batch_in_epoch": 291,
    "theta1": 0.004086,
    "theta2": 0.008743,
    "loss": 0.057588,
    "eta2d_x": 0.355056,
    "eta2d_y": 0.934845,
    "speed": 3.5e-05,
    "heading_rad": 0.02179,
    "drift_mean": 0.01074591,
    "drift_var": 0.0014355273,
    "opacity_var": 0.0009626329,
    "photosphere_crossed": false
  },
  {
    "step": 3104,
    "epoch": 7,
    "batch_in_epoch": 295,
    "theta1": 0.004194,
    "theta2": 0.00884,
    "loss": 0.116519,
    "eta2d_x": 0.384752,
    "eta2d_y": 0.92302,
    "speed": 3.2e-05,
    "heading_rad": 1.161686,
    "drift_mean": 0.01473268,
    "drift_var": 0.0032047288,
    "opacity_var": 0.0020820219,
    "photosphere_crossed": false
  },
  {
    "step": 3108,
    "epoch": 7,
    "batch_in_epoch": 299,
    "theta1": 0.004225,
    "theta2": 0.008876,
    "loss": 0.027274,
    "eta2d_x": 0.391041,
    "eta2d_y": 0.920373,
    "speed": 6e-06,
    "heading_rad": -3.119695,
    "drift_mean": -0.0223113,
    "drift_var": 0.0027369226,
    "opacity_var": 0.0030608987,
    "photosphere_crossed": false
  },
  {
    "step": 3112,
    "epoch": 7,
    "batch_in_epoch": 303,
    "theta1": 0.004187,
    "theta2": 0.008853,
    "loss": 0.038228,
    "eta2d_x": 0.379578,
    "eta2d_y": 0.92516,
    "speed": 2.2e-05,
    "heading_rad": -2.667519,
    "drift_mean": 0.00273847,
    "drift_var": 0.0003059734,
    "opacity_var": 0.002966114,
    "photosphere_crossed": false
  },
  {
    "step": 3116,
    "epoch": 7,
    "batch_in_epoch": 307,
    "theta1": 0.004166,
    "theta2": 0.008833,
    "loss": 0.010354,
    "eta2d_x": 0.374241,
    "eta2d_y": 0.927331,
    "speed": 1.2e-05,
    "heading_rad": -0.236229,
    "drift_mean": -0.00696834,
    "drift_var": 0.0015209581,
    "opacity_var": 0.0008425008,
    "photosphere_crossed": false
  },
  {
    "step": 3120,
    "epoch": 7,
    "batch_in_epoch": 311,
    "theta1": 0.004252,
    "theta2": 0.008775,
    "loss": 0.010468,
    "eta2d_x": 0.419689,
    "eta2d_y": 0.907668,
    "speed": 3.4e-05,
    "heading_rad": -0.441897,
    "drift_mean": 2.847e-05,
    "drift_var": 0.0022008441,
    "opacity_var": 0.0016713986,
    "photosphere_crossed": false
  },
  {
    "step": 3124,
    "epoch": 7,
    "batch_in_epoch": 315,
    "theta1": 0.004329,
    "theta2": 0.008568,
    "loss": 0.016709,
    "eta2d_x": 0.49349,
    "eta2d_y": 0.869752,
    "speed": 6.7e-05,
    "heading_rad": -1.483551,
    "drift_mean": 0.00156009,
    "drift_var": 0.0012323489,
    "opacity_var": 0.0018305824,
    "photosphere_crossed": false
  },
  {
    "step": 3128,
    "epoch": 7,
    "batch_in_epoch": 319,
    "theta1": 0.004247,
    "theta2": 0.008391,
    "loss": 0.031182,
    "eta2d_x": 0.501828,
    "eta2d_y": 0.864968,
    "speed": 3.6e-05,
    "heading_rad": -2.306233,
    "drift_mean": 0.00361821,
    "drift_var": 0.0001481544,
    "opacity_var": 0.0006186015,
    "photosphere_crossed": false
  },
  {
    "step": 3132,
    "epoch": 7,
    "batch_in_epoch": 323,
    "theta1": 0.004186,
    "theta2": 0.008363,
    "loss": 0.042946,
    "eta2d_x": 0.481215,
    "eta2d_y": 0.876603,
    "speed": 1.8e-05,
    "heading_rad": 1.718706,
    "drift_mean": 0.002941,
    "drift_var": 0.0003315431,
    "opacity_var": 0.0002371723,
    "photosphere_crossed": true
  },
  {
    "step": 3136,
    "epoch": 7,
    "batch_in_epoch": 327,
    "theta1": 0.004012,
    "theta2": 0.008317,
    "loss": 0.017719,
    "eta2d_x": 0.404625,
    "eta2d_y": 0.914483,
    "speed": 4.7e-05,
    "heading_rad": -2.840695,
    "drift_mean": -0.00630662,
    "drift_var": 0.0013223536,
    "opacity_var": 0.000798605,
    "photosphere_crossed": false
  },
  {
    "step": 3140,
    "epoch": 7,
    "batch_in_epoch": 331,
    "theta1": 0.003823,
    "theta2": 0.008189,
    "loss": 0.025027,
    "eta2d_x": 0.320169,
    "eta2d_y": 0.947361,
    "speed": 5.4e-05,
    "heading_rad": -2.317229,
    "drift_mean": 0.0018271,
    "drift_var": 0.0001174134,
    "opacity_var": 0.0006601504,
    "photosphere_crossed": false
  },
  {
    "step": 3144,
    "epoch": 7,
    "batch_in_epoch": 335,
    "theta1": 0.003748,
    "theta2": 0.008147,
    "loss": 0.007238,
    "eta2d_x": 0.276994,
    "eta2d_y": 0.960872,
    "speed": 9e-06,
    "heading_rad": 2.567776,
    "drift_mean": -0.00444749,
    "drift_var": 0.0003749931,
    "opacity_var": 0.0002285822,
    "photosphere_crossed": true
  },
  {
    "step": 3148,
    "epoch": 7,
    "batch_in_epoch": 339,
    "theta1": 0.003748,
    "theta2": 0.008188,
    "loss": 0.011325,
    "eta2d_x": 0.269079,
    "eta2d_y": 0.963118,
    "speed": 6e-06,
    "heading_rad": 1.906868,
    "drift_mean": 0.00102179,
    "drift_var": 0.0001165965,
    "opacity_var": 0.0002459214,
    "photosphere_crossed": true
  },
  {
    "step": 3152,
    "epoch": 7,
    "batch_in_epoch": 343,
    "theta1": 0.003761,
    "theta2": 0.008247,
    "loss": 0.037323,
    "eta2d_x": 0.266407,
    "eta2d_y": 0.963861,
    "speed": 1.5e-05,
    "heading_rad": 0.406856,
    "drift_mean": 0.00649959,
    "drift_var": 0.0001532153,
    "opacity_var": 0.000136678,
    "photosphere_crossed": true
  },
  {
    "step": 3156,
    "epoch": 7,
    "batch_in_epoch": 347,
    "theta1": 0.003841,
    "theta2": 0.008258,
    "loss": 0.009141,
    "eta2d_x": 0.316957,
    "eta2d_y": 0.94844,
    "speed": 2.4e-05,
    "heading_rad": 0.005434,
    "drift_mean": -0.00704551,
    "drift_var": 0.0009951709,
    "opacity_var": 0.0005615854,
    "photosphere_crossed": false
  },
  {
    "step": 3160,
    "epoch": 7,
    "batch_in_epoch": 351,
    "theta1": 0.003773,
    "theta2": 0.008218,
    "loss": 0.047668,
    "eta2d_x": 0.280485,
    "eta2d_y": 0.959858,
    "speed": 2.7e-05,
    "heading_rad": -2.339011,
    "drift_mean": 0.00963175,
    "drift_var": 0.0002369793,
    "opacity_var": 0.0006515255,
    "photosphere_crossed": false
  },
  {
    "step": 3164,
    "epoch": 7,
    "batch_in_epoch": 355,
    "theta1": 0.003681,
    "theta2": 0.008122,
    "loss": 0.015666,
    "eta2d_x": 0.232323,
    "eta2d_y": 0.972639,
    "speed": 2.9e-05,
    "heading_rad": -2.406043,
    "drift_mean": -0.00800056,
    "drift_var": 0.0002854346,
    "opacity_var": 0.0004203194,
    "photosphere_crossed": false
  },
  {
    "step": 3168,
    "epoch": 7,
    "batch_in_epoch": 359,
    "theta1": 0.003644,
    "theta2": 0.008016,
    "loss": 0.148969,
    "eta2d_x": 0.222656,
    "eta2d_y": 0.974897,
    "speed": 1.8e-05,
    "heading_rad": -0.912432,
    "drift_mean": 0.03332582,
    "drift_var": 0.0034761897,
    "opacity_var": 0.0021141237,
    "photosphere_crossed": false
  },
  {
    "step": 3172,
    "epoch": 7,
    "batch_in_epoch": 363,
    "theta1": 0.003724,
    "theta2": 0.008009,
    "loss": 0.035362,
    "eta2d_x": 0.290404,
    "eta2d_y": 0.956904,
    "speed": 3.6e-05,
    "heading_rad": 0.020347,
    "drift_mean": -0.0284017,
    "drift_var": 0.0038690246,
    "opacity_var": 0.004112547,
    "photosphere_crossed": false
  },
  {
    "step": 3176,
    "epoch": 7,
    "batch_in_epoch": 367,
    "theta1": 0.003953,
    "theta2": 0.007987,
    "loss": 0.014965,
    "eta2d_x": 0.464253,
    "eta2d_y": 0.885703,
    "speed": 6.1e-05,
    "heading_rad": -0.117685,
    "drift_mean": -0.0050993,
    "drift_var": 0.000245212,
    "opacity_var": 0.0040247463,
    "photosphere_crossed": false
  },
  {
    "step": 3180,
    "epoch": 7,
    "batch_in_epoch": 371,
    "theta1": 0.004167,
    "theta2": 0.007992,
    "loss": 0.019442,
    "eta2d_x": 0.584808,
    "eta2d_y": 0.811172,
    "speed": 5.2e-05,
    "heading_rad": 0.053015,
    "drift_mean": 0.00111923,
    "drift_var": 0.0010828541,
    "opacity_var": 0.0006710487,
    "photosphere_crossed": false
  },
  {
    "step": 3184,
    "epoch": 7,
    "batch_in_epoch": 375,
    "theta1": 0.00435,
    "theta2": 0.007968,
    "loss": 0.027973,
    "eta2d_x": 0.673824,
    "eta2d_y": 0.738892,
    "speed": 4.9e-05,
    "heading_rad": -0.111091,
    "drift_mean": 0.00213287,
    "drift_var": 0.000212863,
    "opacity_var": 0.000586737,
    "photosphere_crossed": false
  },
  {
    "step": 3188,
    "epoch": 7,
    "batch_in_epoch": 379,
    "theta1": 0.004419,
    "theta2": 0.007986,
    "loss": 0.016797,
    "eta2d_x": 0.693138,
    "eta2d_y": 0.720805,
    "speed": 1.4e-05,
    "heading_rad": 0.90032,
    "drift_mean": -0.00279408,
    "drift_var": 9.00511e-05,
    "opacity_var": 0.0001400595,
    "photosphere_crossed": true
  },
  {
    "step": 3192,
    "epoch": 7,
    "batch_in_epoch": 383,
    "theta1": 0.004449,
    "theta2": 0.008124,
    "loss": 0.014963,
    "eta2d_x": 0.659096,
    "eta2d_y": 0.752059,
    "speed": 4.5e-05,
    "heading_rad": 1.424024,
    "drift_mean": -0.00045842,
    "drift_var": 0.0002882717,
    "opacity_var": 0.0001885355,
    "photosphere_crossed": true
  },
  {
    "step": 3196,
    "epoch": 7,
    "batch_in_epoch": 387,
    "theta1": 0.004372,
    "theta2": 0.008265,
    "loss": 0.036836,
    "eta2d_x": 0.588053,
    "eta2d_y": 0.808822,
    "speed": 4.3e-05,
    "heading_rad": 2.35203,
    "drift_mean": 0.00546811,
    "drift_var": 0.0001409702,
    "opacity_var": 0.0002134858,
    "photosphere_crossed": true
  },
  {
    "step": 3200,
    "epoch": 7,
    "batch_in_epoch": 391,
    "theta1": 0.004319,
    "theta2": 0.008411,
    "loss": 0.042248,
    "eta2d_x": 0.527173,
    "eta2d_y": 0.849758,
    "speed": 2.9e-05,
    "heading_rad": 1.743659,
    "drift_mean": 0.001353,
    "drift_var": 0.0003346335,
    "opacity_var": 0.0003187184,
    "photosphere_crossed": false
  },
  {
    "step": 3204,
    "epoch": 7,
    "batch_in_epoch": 395,
    "theta1": 0.004282,
    "theta2": 0.008425,
    "loss": 0.051247,
    "eta2d_x": 0.508256,
    "eta2d_y": 0.861206,
    "speed": 2.5e-05,
    "heading_rad": -2.460362,
    "drift_mean": 0.00224981,
    "drift_var": 0.0001761912,
    "opacity_var": 0.0002274315,
    "photosphere_crossed": true
  },
  {
    "step": 3208,
    "epoch": 7,
    "batch_in_epoch": 399,
    "theta1": 0.004246,
    "theta2": 0.008424,
    "loss": 0.022581,
    "eta2d_x": 0.493099,
    "eta2d_y": 0.869973,
    "speed": 5e-06,
    "heading_rad": 2.182383,
    "drift_mean": -0.00716642,
    "drift_var": 0.0002815653,
    "opacity_var": 0.0002258732,
    "photosphere_crossed": true
  },
  {
    "step": 3212,
    "epoch": 7,
    "batch_in_epoch": 403,
    "theta1": 0.004247,
    "theta2": 0.008475,
    "loss": 0.012666,
    "eta2d_x": 0.480993,
    "eta2d_y": 0.876724,
    "speed": 1e-06,
    "heading_rad": 2.880988,
    "drift_mean": -0.00247878,
    "drift_var": 4.59951e-05,
    "opacity_var": 0.0001587861,
    "photosphere_crossed": true
  },
  {
    "step": 3216,
    "epoch": 7,
    "batch_in_epoch": 407,
    "theta1": 0.004305,
    "theta2": 0.008435,
    "loss": 0.013498,
    "eta2d_x": 0.515222,
    "eta2d_y": 0.857057,
    "speed": 2.9e-05,
    "heading_rad": -0.830569,
    "drift_mean": 0.0002079,
    "drift_var": 0.0004579324,
    "opacity_var": 0.000247505,
    "photosphere_crossed": true
  },
  {
    "step": 3220,
    "epoch": 7,
    "batch_in_epoch": 411,
    "theta1": 0.004324,
    "theta2": 0.008414,
    "loss": 0.053393,
    "eta2d_x": 0.52828,
    "eta2d_y": 0.84907,
    "speed": 5e-06,
    "heading_rad": 2.28717,
    "drift_mean": 0.00997377,
    "drift_var": 7.40564e-05,
    "opacity_var": 0.0002582724,
    "photosphere_crossed": true
  },
  {
    "step": 3224,
    "epoch": 7,
    "batch_in_epoch": 415,
    "theta1": 0.004324,
    "theta2": 0.008451,
    "loss": 0.015501,
    "eta2d_x": 0.518843,
    "eta2d_y": 0.854869,
    "speed": 1.4e-05,
    "heading_rad": 1.504275,
    "drift_mean": -0.0094731,
    "drift_var": 0.0005950449,
    "opacity_var": 0.0004563591,
    "photosphere_crossed": false
  },
  {
    "step": 3228,
    "epoch": 7,
    "batch_in_epoch": 419,
    "theta1": 0.004387,
    "theta2": 0.008507,
    "loss": 0.007941,
    "eta2d_x": 0.530399,
    "eta2d_y": 0.847748,
    "speed": 2.8e-05,
    "heading_rad": 0.521718,
    "drift_mean": -0.00188987,
    "drift_var": 0.0017437146,
    "opacity_var": 0.0010609376,
    "photosphere_crossed": false
  },
  {
    "step": 3232,
    "epoch": 7,
    "batch_in_epoch": 423,
    "theta1": 0.004499,
    "theta2": 0.008592,
    "loss": 0.022716,
    "eta2d_x": 0.55134,
    "eta2d_y": 0.834281,
    "speed": 3e-05,
    "heading_rad": 0.593408,
    "drift_mean": 0.0036936,
    "drift_var": 0.0004500384,
    "opacity_var": 0.0009819685,
    "photosphere_crossed": false
  },
  {
    "step": 3236,
    "epoch": 7,
    "batch_in_epoch": 427,
    "theta1": 0.004624,
    "theta2": 0.008731,
    "loss": 0.020365,
    "eta2d_x": 0.561698,
    "eta2d_y": 0.827342,
    "speed": 6.5e-05,
    "heading_rad": 1.014954,
    "drift_mean": -0.00058753,
    "drift_var": 0.0001550655,
    "opacity_var": 0.0007892841,
    "photosphere_crossed": false
  },
  {
    "step": 3240,
    "epoch": 7,
    "batch_in_epoch": 431,
    "theta1": 0.004676,
    "theta2": 0.008888,
    "loss": 0.033961,
    "eta2d_x": 0.545801,
    "eta2d_y": 0.837915,
    "speed": 2.3e-05,
    "heading_rad": 1.349979,
    "drift_mean": 0.00339879,
    "drift_var": 0.000248219,
    "opacity_var": 0.0001908818,
    "photosphere_crossed": true
  },
  {
    "step": 3244,
    "epoch": 7,
    "batch_in_epoch": 435,
    "theta1": 0.00473,
    "theta2": 0.008941,
    "loss": 0.048578,
    "eta2d_x": 0.551452,
    "eta2d_y": 0.834206,
    "speed": 3.2e-05,
    "heading_rad": 1.012136,
    "drift_mean": 0.00365444,
    "drift_var": 0.0005005263,
    "opacity_var": 0.000354343,
    "photosphere_crossed": false
  },
  {
    "step": 3248,
    "epoch": 7,
    "batch_in_epoch": 439,
    "theta1": 0.004757,
    "theta2": 0.00905,
    "loss": 0.005696,
    "eta2d_x": 0.539149,
    "eta2d_y": 0.84221,
    "speed": 2.8e-05,
    "heading_rad": 1.508195,
    "drift_mean": -0.01072059,
    "drift_var": 0.0009075695,
    "opacity_var": 0.0006741065,
    "photosphere_crossed": false
  },
  {
    "step": 3252,
    "epoch": 7,
    "batch_in_epoch": 443,
    "theta1": 0.004779,
    "theta2": 0.009174,
    "loss": 0.016654,
    "eta2d_x": 0.523469,
    "eta2d_y": 0.852044,
    "speed": 2.7e-05,
    "heading_rad": 1.431495,
    "drift_mean": 0.00273938,
    "drift_var": 0.0010093612,
    "opacity_var": 0.0010741214,
    "photosphere_crossed": false
  },
  {
    "step": 3256,
    "epoch": 7,
    "batch_in_epoch": 447,
    "theta1": 0.004805,
    "theta2": 0.009135,
    "loss": 0.02527,
    "eta2d_x": 0.537194,
    "eta2d_y": 0.843458,
    "speed": 2.5e-05,
    "heading_rad": -0.628415,
    "drift_mean": 0.00215416,
    "drift_var": 0.0002091235,
    "opacity_var": 0.0007578851,
    "photosphere_crossed": false
  },
  {
    "step": 3260,
    "epoch": 7,
    "batch_in_epoch": 451,
    "theta1": 0.004874,
    "theta2": 0.00901,
    "loss": 0.020914,
    "eta2d_x": 0.57796,
    "eta2d_y": 0.816065,
    "speed": 4.5e-05,
    "heading_rad": -0.830288,
    "drift_mean": -0.00108911,
    "drift_var": 0.0001936359,
    "opacity_var": 0.0002520767,
    "photosphere_crossed": true
  },
  {
    "step": 3264,
    "epoch": 7,
    "batch_in_epoch": 455,
    "theta1": 0.004965,
    "theta2": 0.008909,
    "loss": 0.037973,
    "eta2d_x": 0.619828,
    "eta2d_y": 0.784738,
    "speed": 3.7e-05,
    "heading_rad": -1.241486,
    "drift_mean": 0.00426479,
    "drift_var": 0.0002764756,
    "opacity_var": 0.0002434555,
    "photosphere_crossed": true
  },
  {
    "step": 3268,
    "epoch": 7,
    "batch_in_epoch": 459,
    "theta1": 0.004975,
    "theta2": 0.008763,
    "loss": 0.023346,
    "eta2d_x": 0.651145,
    "eta2d_y": 0.758953,
    "speed": 3.9e-05,
    "heading_rad": -1.528908,
    "drift_mean": -0.00365677,
    "drift_var": 0.0003102562,
    "opacity_var": 0.0002750002,
    "photosphere_crossed": true
  },
  {
    "step": 3272,
    "epoch": 7,
    "batch_in_epoch": 463,
    "theta1": 0.00499,
    "theta2": 0.008624,
    "loss": 0.08609,
    "eta2d_x": 0.683563,
    "eta2d_y": 0.729892,
    "speed": 4.6e-05,
    "heading_rad": -1.59225,
    "drift_mean": 0.01568597,
    "drift_var": 0.0014517581,
    "opacity_var": 0.0009331665,
    "photosphere_crossed": false
  },
  {
    "step": 3276,
    "epoch": 7,
    "batch_in_epoch": 467,
    "theta1": 0.00497,
    "theta2": 0.008583,
    "loss": 0.022011,
    "eta2d_x": 0.687858,
    "eta2d_y": 0.725845,
    "speed": 1.1e-05,
    "heading_rad": 1.434929,
    "drift_mean": -0.01601956,
    "drift_var": 0.0007421898,
    "opacity_var": 0.001200134,
    "photosphere_crossed": false
  },
  {
    "step": 3280,
    "epoch": 8,
    "batch_in_epoch": 3,
    "theta1": 0.004956,
    "theta2": 0.00854,
    "loss": 0.008048,
    "eta2d_x": 0.693985,
    "eta2d_y": 0.71999,
    "speed": 1.9e-05,
    "heading_rad": -1.908779,
    "drift_mean": -0.00349083,
    "drift_var": 0.0001685829,
    "opacity_var": 0.0012599292,
    "photosphere_crossed": false
  },
  {
    "step": 3284,
    "epoch": 8,
    "batch_in_epoch": 7,
    "theta1": 0.004931,
    "theta2": 0.008519,
    "loss": 0.013267,
    "eta2d_x": 0.692864,
    "eta2d_y": 0.721068,
    "speed": 6e-06,
    "heading_rad": 1.217093,
    "drift_mean": 0.00130484,
    "drift_var": 3.71724e-05,
    "opacity_var": 9.67195e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3288,
    "epoch": 8,
    "batch_in_epoch": 11,
    "theta1": 0.004955,
    "theta2": 0.008588,
    "loss": 0.015879,
    "eta2d_x": 0.683104,
    "eta2d_y": 0.730321,
    "speed": 1.2e-05,
    "heading_rad": 1.507729,
    "drift_mean": 0.00065297,
    "drift_var": 3.75143e-05,
    "opacity_var": 7.165e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3292,
    "epoch": 8,
    "batch_in_epoch": 15,
    "theta1": 0.005048,
    "theta2": 0.008614,
    "loss": 0.027121,
    "eta2d_x": 0.698452,
    "eta2d_y": 0.715657,
    "speed": 2.3e-05,
    "heading_rad": 0.446901,
    "drift_mean": 0.00281036,
    "drift_var": 6.57758e-05,
    "opacity_var": 5.52361e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3296,
    "epoch": 8,
    "batch_in_epoch": 19,
    "theta1": 0.005045,
    "theta2": 0.008617,
    "loss": 0.006504,
    "eta2d_x": 0.697334,
    "eta2d_y": 0.716747,
    "speed": 3e-06,
    "heading_rad": -2.977456,
    "drift_mean": -0.0051543,
    "drift_var": 0.000265722,
    "opacity_var": 0.0001683725,
    "photosphere_crossed": true
  },
  {
    "step": 3300,
    "epoch": 8,
    "batch_in_epoch": 23,
    "theta1": 0.005024,
    "theta2": 0.008651,
    "loss": 0.042858,
    "eta2d_x": 0.685442,
    "eta2d_y": 0.728127,
    "speed": 2.5e-05,
    "heading_rad": 1.671268,
    "drift_mean": 0.00908855,
    "drift_var": 0.0001365907,
    "opacity_var": 0.0002259451,
    "photosphere_crossed": true
  },
  {
    "step": 3304,
    "epoch": 8,
    "batch_in_epoch": 27,
    "theta1": 0.004952,
    "theta2": 0.008761,
    "loss": 0.01046,
    "eta2d_x": 0.645915,
    "eta2d_y": 0.763409,
    "speed": 2.9e-05,
    "heading_rad": 1.97992,
    "drift_mean": -0.00809949,
    "drift_var": 0.0003054838,
    "opacity_var": 0.0002622731,
    "photosphere_crossed": true
  },
  {
    "step": 3308,
    "epoch": 8,
    "batch_in_epoch": 31,
    "theta1": 0.004907,
    "theta2": 0.008799,
    "loss": 0.014534,
    "eta2d_x": 0.627215,
    "eta2d_y": 0.778847,
    "speed": 7e-06,
    "heading_rad": -2.780287,
    "drift_mean": 0.00101851,
    "drift_var": 0.0001580939,
    "opacity_var": 0.0003276685,
    "photosphere_crossed": false
  },
  {
    "step": 3312,
    "epoch": 8,
    "batch_in_epoch": 35,
    "theta1": 0.004996,
    "theta2": 0.008869,
    "loss": 0.004709,
    "eta2d_x": 0.634934,
    "eta2d_y": 0.772566,
    "speed": 1.9e-05,
    "heading_rad": 0.545291,
    "drift_mean": -0.00245615,
    "drift_var": 0.0003027758,
    "opacity_var": 0.0002231038,
    "photosphere_crossed": true
  },
  {
    "step": 3316,
    "epoch": 8,
    "batch_in_epoch": 39,
    "theta1": 0.005059,
    "theta2": 0.008941,
    "loss": 0.008871,
    "eta2d_x": 0.635965,
    "eta2d_y": 0.771718,
    "speed": 1.8e-05,
    "heading_rad": 0.551434,
    "drift_mean": 0.00104048,
    "drift_var": 1.6811e-05,
    "opacity_var": 0.0001511399,
    "photosphere_crossed": true
  },
  {
    "step": 3320,
    "epoch": 8,
    "batch_in_epoch": 43,
    "theta1": 0.00514,
    "theta2": 0.008919,
    "loss": 0.020605,
    "eta2d_x": 0.657746,
    "eta2d_y": 0.753239,
    "speed": 3e-05,
    "heading_rad": -0.522053,
    "drift_mean": 0.00293354,
    "drift_var": 0.0001086312,
    "opacity_var": 5.79738e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3324,
    "epoch": 8,
    "batch_in_epoch": 47,
    "theta1": 0.005198,
    "theta2": 0.008858,
    "loss": 0.010589,
    "eta2d_x": 0.68137,
    "eta2d_y": 0.731939,
    "speed": 1.9e-05,
    "heading_rad": -1.437721,
    "drift_mean": -0.002504,
    "drift_var": 4.83167e-05,
    "opacity_var": 7.64871e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3328,
    "epoch": 8,
    "batch_in_epoch": 51,
    "theta1": 0.005134,
    "theta2": 0.008788,
    "loss": 0.028215,
    "eta2d_x": 0.681484,
    "eta2d_y": 0.731833,
    "speed": 3e-05,
    "heading_rad": -2.53023,
    "drift_mean": 0.00440629,
    "drift_var": 0.0001533567,
    "opacity_var": 0.0001102705,
    "photosphere_crossed": true
  },
  {
    "step": 3332,
    "epoch": 8,
    "batch_in_epoch": 55,
    "theta1": 0.005072,
    "theta2": 0.008784,
    "loss": 0.010023,
    "eta2d_x": 0.669019,
    "eta2d_y": 0.743245,
    "speed": 1.1e-05,
    "heading_rad": 2.834411,
    "drift_mean": -0.00454792,
    "drift_var": 0.0004227738,
    "opacity_var": 0.0002742718,
    "photosphere_crossed": true
  },
  {
    "step": 3336,
    "epoch": 8,
    "batch_in_epoch": 59,
    "theta1": 0.005107,
    "theta2": 0.008764,
    "loss": 0.018439,
    "eta2d_x": 0.680559,
    "eta2d_y": 0.732693,
    "speed": 1.1e-05,
    "heading_rad": -0.09329,
    "drift_mean": 0.00210405,
    "drift_var": 0.0001296419,
    "opacity_var": 0.0002673945,
    "photosphere_crossed": true
  },
  {
    "step": 3340,
    "epoch": 8,
    "batch_in_epoch": 63,
    "theta1": 0.005038,
    "theta2": 0.008818,
    "loss": 0.014091,
    "eta2d_x": 0.654522,
    "eta2d_y": 0.756043,
    "speed": 2.8e-05,
    "heading_rad": 2.369588,
    "drift_mean": -0.00108708,
    "drift_var": 0.0001732916,
    "opacity_var": 0.0001510434,
    "photosphere_crossed": true
  },
  {
    "step": 3344,
    "epoch": 8,
    "batch_in_epoch": 67,
    "theta1": 0.004995,
    "theta2": 0.008916,
    "loss": 0.018061,
    "eta2d_x": 0.625613,
    "eta2d_y": 0.780134,
    "speed": 1.3e-05,
    "heading_rad": 1.807026,
    "drift_mean": 0.00099259,
    "drift_var": 2.54034e-05,
    "opacity_var": 0.0001023575,
    "photosphere_crossed": true
  },
  {
    "step": 3348,
    "epoch": 8,
    "batch_in_epoch": 71,
    "theta1": 0.004961,
    "theta2": 0.008945,
    "loss": 0.036839,
    "eta2d_x": 0.612122,
    "eta2d_y": 0.790764,
    "speed": 1.8e-05,
    "heading_rad": -3.128606,
    "drift_mean": 0.00469436,
    "drift_var": 0.0001709503,
    "opacity_var": 9.42481e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3352,
    "epoch": 8,
    "batch_in_epoch": 75,
    "theta1": 0.004857,
    "theta2": 0.008894,
    "loss": 0.037359,
    "eta2d_x": 0.595222,
    "eta2d_y": 0.803561,
    "speed": 4.4e-05,
    "heading_rad": -2.600128,
    "drift_mean": 0.00013009,
    "drift_var": 0.000383435,
    "opacity_var": 0.0002554695,
    "photosphere_crossed": true
  },
  {
    "step": 3356,
    "epoch": 8,
    "batch_in_epoch": 79,
    "theta1": 0.004741,
    "theta2": 0.008778,
    "loss": 0.010044,
    "eta2d_x": 0.586787,
    "eta2d_y": 0.809742,
    "speed": 3.6e-05,
    "heading_rad": -2.185057,
    "drift_mean": -0.00682876,
    "drift_var": 0.0002866061,
    "opacity_var": 0.000352482,
    "photosphere_crossed": false
  },
  {
    "step": 3360,
    "epoch": 8,
    "batch_in_epoch": 83,
    "theta1": 0.004743,
    "theta2": 0.008548,
    "loss": 0.003319,
    "eta2d_x": 0.637915,
    "eta2d_y": 0.770107,
    "speed": 6.1e-05,
    "heading_rad": -1.500733,
    "drift_mean": -0.00168125,
    "drift_var": 0.0006001542,
    "opacity_var": 0.0004094259,
    "photosphere_crossed": false
  },
  {
    "step": 3364,
    "epoch": 8,
    "batch_in_epoch": 87,
    "theta1": 0.004795,
    "theta2": 0.008384,
    "loss": 0.01071,
    "eta2d_x": 0.691264,
    "eta2d_y": 0.722603,
    "speed": 3.1e-05,
    "heading_rad": -1.285911,
    "drift_mean": 0.00184768,
    "drift_var": 0.0011197588,
    "opacity_var": 0.0007958408,
    "photosphere_crossed": false
  },
  {
    "step": 3368,
    "epoch": 8,
    "batch_in_epoch": 91,
    "theta1": 0.004864,
    "theta2": 0.008268,
    "loss": 0.095908,
    "eta2d_x": 0.73712,
    "eta2d_y": 0.675762,
    "speed": 4.3e-05,
    "heading_rad": -0.909683,
    "drift_mean": 0.02129949,
    "drift_var": 0.0019349659,
    "opacity_var": 0.0014873679,
    "photosphere_crossed": false
  },
  {
    "step": 3372,
    "epoch": 8,
    "batch_in_epoch": 95,
    "theta1": 0.004989,
    "theta2": 0.008275,
    "loss": 0.010324,
    "eta2d_x": 0.761985,
    "eta2d_y": 0.647594,
    "speed": 5.7e-05,
    "heading_rad": 0.39516,
    "drift_mean": -0.02139596,
    "drift_var": 0.0011409366,
    "opacity_var": 0.0017765359,
    "photosphere_crossed": false
  },
  {
    "step": 3376,
    "epoch": 8,
    "batch_in_epoch": 99,
    "theta1": 0.005136,
    "theta2": 0.008388,
    "loss": 0.010803,
    "eta2d_x": 0.764627,
    "eta2d_y": 0.644473,
    "speed": 4.2e-05,
    "heading_rad": 1.027358,
    "drift_mean": 0.00011989,
    "drift_var": 0.0003746713,
    "opacity_var": 0.0018001659,
    "photosphere_crossed": false
  },
  {
    "step": 3380,
    "epoch": 8,
    "batch_in_epoch": 103,
    "theta1": 0.005232,
    "theta2": 0.008548,
    "loss": 0.041832,
    "eta2d_x": 0.748461,
    "eta2d_y": 0.663179,
    "speed": 5.8e-05,
    "heading_rad": 1.032388,
    "drift_mean": 0.00775725,
    "drift_var": 4.66091e-05,
    "opacity_var": 0.0002118669,
    "photosphere_crossed": true
  },
  {
    "step": 3384,
    "epoch": 8,
    "batch_in_epoch": 107,
    "theta1": 0.0053,
    "theta2": 0.008689,
    "loss": 0.005698,
    "eta2d_x": 0.732961,
    "eta2d_y": 0.680271,
    "speed": 2.3e-05,
    "heading_rad": 1.178577,
    "drift_mean": -0.00903351,
    "drift_var": 0.0002893562,
    "opacity_var": 0.000290167,
    "photosphere_crossed": true
  },
  {
    "step": 3388,
    "epoch": 8,
    "batch_in_epoch": 111,
    "theta1": 0.005272,
    "theta2": 0.00877,
    "loss": 0.028123,
    "eta2d_x": 0.712355,
    "eta2d_y": 0.701819,
    "speed": 1.9e-05,
    "heading_rad": 2.327017,
    "drift_mean": 0.00560616,
    "drift_var": 9.76005e-05,
    "opacity_var": 0.0002545664,
    "photosphere_crossed": true
  },
  {
    "step": 3392,
    "epoch": 8,
    "batch_in_epoch": 115,
    "theta1": 0.005228,
    "theta2": 0.008806,
    "loss": 0.005261,
    "eta2d_x": 0.697161,
    "eta2d_y": 0.716915,
    "speed": 7e-06,
    "heading_rad": -3.093604,
    "drift_mean": -0.00571543,
    "drift_var": 9.97706e-05,
    "opacity_var": 0.0001353414,
    "photosphere_crossed": true
  },
  {
    "step": 3396,
    "epoch": 8,
    "batch_in_epoch": 119,
    "theta1": 0.005213,
    "theta2": 0.008708,
    "loss": 0.011563,
    "eta2d_x": 0.713068,
    "eta2d_y": 0.701094,
    "speed": 3.6e-05,
    "heading_rad": -1.533632,
    "drift_mean": 0.00157555,
    "drift_var": 0.0017504917,
    "opacity_var": 0.0008530115,
    "photosphere_crossed": false
  },
  {
    "step": 3400,
    "epoch": 8,
    "batch_in_epoch": 123,
    "theta1": 0.005225,
    "theta2": 0.008605,
    "loss": 0.040724,
    "eta2d_x": 0.735806,
    "eta2d_y": 0.677192,
    "speed": 2.5e-05,
    "heading_rad": -1.498268,
    "drift_mean": 0.00729016,
    "drift_var": 0.0001451217,
    "opacity_var": 0.0008681819,
    "photosphere_crossed": false
  },
  {
    "step": 3404,
    "epoch": 8,
    "batch_in_epoch": 127,
    "theta1": 0.005195,
    "theta2": 0.008452,
    "loss": 0.01409,
    "eta2d_x": 0.761691,
    "eta2d_y": 0.64794,
    "speed": 3.2e-05,
    "heading_rad": -1.729826,
    "drift_mean": -0.0066585,
    "drift_var": 2.07007e-05,
    "opacity_var": 0.0001172714,
    "photosphere_crossed": true
  },
  {
    "step": 3408,
    "epoch": 8,
    "batch_in_epoch": 131,
    "theta1": 0.005168,
    "theta2": 0.008406,
    "loss": 0.007551,
    "eta2d_x": 0.766597,
    "eta2d_y": 0.642128,
    "speed": 7e-06,
    "heading_rad": 2.177977,
    "drift_mean": -0.00163473,
    "drift_var": 0.0002132421,
    "opacity_var": 0.0002045263,
    "photosphere_crossed": true
  },
  {
    "step": 3412,
    "epoch": 8,
    "batch_in_epoch": 135,
    "theta1": 0.005121,
    "theta2": 0.008334,
    "loss": 0.009803,
    "eta2d_x": 0.773661,
    "eta2d_y": 0.633599,
    "speed": 3.2e-05,
    "heading_rad": -1.936546,
    "drift_mean": 0.0005629,
    "drift_var": 0.0001165738,
    "opacity_var": 0.0001476613,
    "photosphere_crossed": true
  },
  {
    "step": 3416,
    "epoch": 8,
    "batch_in_epoch": 139,
    "theta1": 0.005044,
    "theta2": 0.008322,
    "loss": 0.01839,
    "eta2d_x": 0.762085,
    "eta2d_y": 0.647477,
    "speed": 1.9e-05,
    "heading_rad": 2.664149,
    "drift_mean": 0.00214697,
    "drift_var": 0.0001126213,
    "opacity_var": 0.0001565587,
    "photosphere_crossed": true
  },
  {
    "step": 3420,
    "epoch": 8,
    "batch_in_epoch": 143,
    "theta1": 0.004843,
    "theta2": 0.008262,
    "loss": 0.015314,
    "eta2d_x": 0.733706,
    "eta2d_y": 0.679467,
    "speed": 5.6e-05,
    "heading_rad": -2.653457,
    "drift_mean": -0.00076919,
    "drift_var": 0.000119488,
    "opacity_var": 0.0001057281,
    "photosphere_crossed": true
  },
  {
    "step": 3424,
    "epoch": 8,
    "batch_in_epoch": 147,
    "theta1": 0.00469,
    "theta2": 0.008183,
    "loss": 0.081227,
    "eta2d_x": 0.715907,
    "eta2d_y": 0.698196,
    "speed": 3e-05,
    "heading_rad": 2.830431,
    "drift_mean": 0.01647825,
    "drift_var": 0.0010477181,
    "opacity_var": 0.0005855254,
    "photosphere_crossed": false
  },
  {
    "step": 3428,
    "epoch": 8,
    "batch_in_epoch": 151,
    "theta1": 0.004663,
    "theta2": 0.008322,
    "loss": 0.025749,
    "eta2d_x": 0.671041,
    "eta2d_y": 0.74142,
    "speed": 4.1e-05,
    "heading_rad": 1.279883,
    "drift_mean": -0.0138694,
    "drift_var": 0.0007857035,
    "opacity_var": 0.0010304594,
    "photosphere_crossed": false
  },
  {
    "step": 3432,
    "epoch": 8,
    "batch_in_epoch": 155,
    "theta1": 0.004707,
    "theta2": 0.008403,
    "loss": 0.019883,
    "eta2d_x": 0.66295,
    "eta2d_y": 0.748664,
    "speed": 1.1e-05,
    "heading_rad": 0.384172,
    "drift_mean": -0.00146651,
    "drift_var": 9.63733e-05,
    "opacity_var": 0.0010561016,
    "photosphere_crossed": false
  },
  {
    "step": 3436,
    "epoch": 8,
    "batch_in_epoch": 159,
    "theta1": 0.004777,
    "theta2": 0.008367,
    "loss": 0.017232,
    "eta2d_x": 0.690684,
    "eta2d_y": 0.723157,
    "speed": 2.1e-05,
    "heading_rad": 0.1008,
    "drift_mean": -0.00066286,
    "drift_var": 0.0002946597,
    "opacity_var": 0.00026004,
    "photosphere_crossed": true
  },
  {
    "step": 3440,
    "epoch": 8,
    "batch_in_epoch": 163,
    "theta1": 0.004917,
    "theta2": 0.008362,
    "loss": 0.030267,
    "eta2d_x": 0.726082,
    "eta2d_y": 0.687608,
    "speed": 3.6e-05,
    "heading_rad": -0.22666,
    "drift_mean": 0.00325873,
    "drift_var": 0.0053575382,
    "opacity_var": 0.0025176401,
    "photosphere_crossed": false
  },
  {
    "step": 3444,
    "epoch": 8,
    "batch_in_epoch": 167,
    "theta1": 0.004988,
    "theta2": 0.008327,
    "loss": 0.020631,
    "eta2d_x": 0.749569,
    "eta2d_y": 0.661926,
    "speed": 1e-05,
    "heading_rad": -0.89055,
    "drift_mean": -0.00240891,
    "drift_var": 0.0001433382,
    "opacity_var": 0.0024644586,
    "photosphere_crossed": false
  },
  {
    "step": 3448,
    "epoch": 8,
    "batch_in_epoch": 171,
    "theta1": 0.005105,
    "theta2": 0.008327,
    "loss": 0.030895,
    "eta2d_x": 0.772357,
    "eta2d_y": 0.635188,
    "speed": 3.2e-05,
    "heading_rad": 0.310206,
    "drift_mean": 0.00256599,
    "drift_var": 0.0002126431,
    "opacity_var": 0.0001637665,
    "photosphere_crossed": true
  },
  {
    "step": 3452,
    "epoch": 8,
    "batch_in_epoch": 175,
    "theta1": 0.005077,
    "theta2": 0.008338,
    "loss": 0.005782,
    "eta2d_x": 0.764829,
    "eta2d_y": 0.644234,
    "speed": 1.4e-05,
    "heading_rad": -2.954575,
    "drift_mean": -0.00627831,
    "drift_var": 3.12349e-05,
    "opacity_var": 0.0001509931,
    "photosphere_crossed": true
  },
  {
    "step": 3456,
    "epoch": 8,
    "batch_in_epoch": 179,
    "theta1": 0.005085,
    "theta2": 0.008343,
    "loss": 0.009115,
    "eta2d_x": 0.765046,
    "eta2d_y": 0.643975,
    "speed": 1.2e-05,
    "heading_rad": -0.341768,
    "drift_mean": 0.00083338,
    "drift_var": 0.0004770388,
    "opacity_var": 0.0002454038,
    "photosphere_crossed": true
  },
  {
    "step": 3460,
    "epoch": 8,
    "batch_in_epoch": 183,
    "theta1": 0.005105,
    "theta2": 0.008302,
    "loss": 0.008562,
    "eta2d_x": 0.777942,
    "eta2d_y": 0.628336,
    "speed": 1.9e-05,
    "heading_rad": -1.499785,
    "drift_mean": -0.00013839,
    "drift_var": 0.0004803808,
    "opacity_var": 0.0004380535,
    "photosphere_crossed": false
  },
  {
    "step": 3464,
    "epoch": 8,
    "batch_in_epoch": 187,
    "theta1": 0.005062,
    "theta2": 0.008181,
    "loss": 0.01204,
    "eta2d_x": 0.797591,
    "eta2d_y": 0.603199,
    "speed": 3.3e-05,
    "heading_rad": -2.190454,
    "drift_mean": 0.00086951,
    "drift_var": 0.0005603612,
    "opacity_var": 0.0006001686,
    "photosphere_crossed": false
  },
  {
    "step": 3468,
    "epoch": 8,
    "batch_in_epoch": 191,
    "theta1": 0.004958,
    "theta2": 0.008168,
    "loss": 0.08722,
    "eta2d_x": 0.781544,
    "eta2d_y": 0.62385,
    "speed": 3.5e-05,
    "heading_rad": 2.618808,
    "drift_mean": 0.01879512,
    "drift_var": 0.0013859967,
    "opacity_var": 0.0010459439,
    "photosphere_crossed": false
  },
  {
    "step": 3472,
    "epoch": 8,
    "batch_in_epoch": 195,
    "theta1": 0.004849,
    "theta2": 0.008222,
    "loss": 0.019691,
    "eta2d_x": 0.745077,
    "eta2d_y": 0.666978,
    "speed": 3.3e-05,
    "heading_rad": 3.084601,
    "drift_mean": -0.01688225,
    "drift_var": 0.0014061215,
    "opacity_var": 0.0015284037,
    "photosphere_crossed": false
  },
  {
    "step": 3476,
    "epoch": 8,
    "batch_in_epoch": 199,
    "theta1": 0.004763,
    "theta2": 0.008293,
    "loss": 0.05235,
    "eta2d_x": 0.706217,
    "eta2d_y": 0.707995,
    "speed": 3.8e-05,
    "heading_rad": 2.080852,
    "drift_mean": 0.00816475,
    "drift_var": 0.0003080908,
    "opacity_var": 0.0016486997,
    "photosphere_crossed": false
  },
  {
    "step": 3480,
    "epoch": 8,
    "batch_in_epoch": 203,
    "theta1": 0.004791,
    "theta2": 0.008433,
    "loss": 0.030001,
    "eta2d_x": 0.678194,
    "eta2d_y": 0.734882,
    "speed": 5.6e-05,
    "heading_rad": 0.744934,
    "drift_mean": -0.00558729,
    "drift_var": 0.0006787848,
    "opacity_var": 0.0004901042,
    "photosphere_crossed": false
  },
  {
    "step": 3484,
    "epoch": 8,
    "batch_in_epoch": 207,
    "theta1": 0.004951,
    "theta2": 0.008494,
    "loss": 0.035914,
    "eta2d_x": 0.702983,
    "eta2d_y": 0.711207,
    "speed": 3.7e-05,
    "heading_rad": -0.40727,
    "drift_mean": 0.00147826,
    "drift_var": 0.0001053383,
    "opacity_var": 0.0005052714,
    "photosphere_crossed": false
  },
  {
    "step": 3488,
    "epoch": 8,
    "batch_in_epoch": 211,
    "theta1": 0.005037,
    "theta2": 0.008405,
    "loss": 0.018952,
    "eta2d_x": 0.741828,
    "eta2d_y": 0.67059,
    "speed": 4.5e-05,
    "heading_rad": -1.035239,
    "drift_mean": -0.0042405,
    "drift_var": 0.0002421882,
    "opacity_var": 0.0002077286,
    "photosphere_crossed": true
  },
  {
    "step": 3492,
    "epoch": 8,
    "batch_in_epoch": 215,
    "theta1": 0.005096,
    "theta2": 0.008298,
    "loss": 0.009208,
    "eta2d_x": 0.777238,
    "eta2d_y": 0.629207,
    "speed": 2.5e-05,
    "heading_rad": -0.879939,
    "drift_mean": -0.00243593,
    "drift_var": 6.19501e-05,
    "opacity_var": 0.0001696839,
    "photosphere_crossed": true
  },
  {
    "step": 3496,
    "epoch": 8,
    "batch_in_epoch": 219,
    "theta1": 0.005149,
    "theta2": 0.008245,
    "loss": 0.017056,
    "eta2d_x": 0.798092,
    "eta2d_y": 0.602536,
    "speed": 3e-05,
    "heading_rad": 0.264779,
    "drift_mean": 0.00196189,
    "drift_var": 0.0005215958,
    "opacity_var": 0.0002656777,
    "photosphere_crossed": true
  },
  {
    "step": 3500,
    "epoch": 8,
    "batch_in_epoch": 223,
    "theta1": 0.005218,
    "theta2": 0.00826,
    "loss": 0.044613,
    "eta2d_x": 0.805866,
    "eta2d_y": 0.592098,
    "speed": 3e-06,
    "heading_rad": -0.286823,
    "drift_mean": 0.00688919,
    "drift_var": 0.006264415,
    "opacity_var": 0.0030247639,
    "photosphere_crossed": false
  },
  {
    "step": 3504,
    "epoch": 8,
    "batch_in_epoch": 227,
    "theta1": 0.0052,
    "theta2": 0.008248,
    "loss": 0.028485,
    "eta2d_x": 0.805612,
    "eta2d_y": 0.592443,
    "speed": 8e-06,
    "heading_rad": 2.923612,
    "drift_mean": -0.00403201,
    "drift_var": 0.0003490211,
    "opacity_var": 0.0029707811,
    "photosphere_crossed": false
  },
  {
    "step": 3508,
    "epoch": 8,
    "batch_in_epoch": 231,
    "theta1": 0.005203,
    "theta2": 0.008223,
    "loss": 0.004692,
    "eta2d_x": 0.811301,
    "eta2d_y": 0.584629,
    "speed": 7e-06,
    "heading_rad": -2.152924,
    "drift_mean": -0.00594802,
    "drift_var": 0.0003947944,
    "opacity_var": 0.0003315669,
    "photosphere_crossed": false
  },
  {
    "step": 3512,
    "epoch": 8,
    "batch_in_epoch": 235,
    "theta1": 0.00516,
    "theta2": 0.008146,
    "loss": 0.027438,
    "eta2d_x": 0.821431,
    "eta2d_y": 0.570308,
    "speed": 2.3e-05,
    "heading_rad": -2.150374,
    "drift_mean": 0.00568646,
    "drift_var": 0.0007392505,
    "opacity_var": 0.0005433463,
    "photosphere_crossed": false
  },
  {
    "step": 3516,
    "epoch": 8,
    "batch_in_epoch": 239,
    "theta1": 0.005142,
    "theta2": 0.007981,
    "loss": 0.003516,
    "eta2d_x": 0.85487,
    "eta2d_y": 0.518841,
    "speed": 5.9e-05,
    "heading_rad": -1.522374,
    "drift_mean": -0.00598048,
    "drift_var": 0.0009101197,
    "opacity_var": 0.0008653763,
    "photosphere_crossed": false
  },
  {
    "step": 3520,
    "epoch": 8,
    "batch_in_epoch": 243,
    "theta1": 0.005121,
    "theta2": 0.007786,
    "loss": 0.010208,
    "eta2d_x": 0.893831,
    "eta2d_y": 0.448403,
    "speed": 3.9e-05,
    "heading_rad": -1.866114,
    "drift_mean": 0.00167297,
    "drift_var": 0.0004148416,
    "opacity_var": 0.000638645,
    "photosphere_crossed": false
  },
  {
    "step": 3524,
    "epoch": 8,
    "batch_in_epoch": 247,
    "theta1": 0.005159,
    "theta2": 0.00771,
    "loss": 0.001676,
    "eta2d_x": 0.912779,
    "eta2d_y": 0.408453,
    "speed": 1.8e-05,
    "heading_rad": -0.707487,
    "drift_mean": -0.00213302,
    "drift_var": 2.74369e-05,
    "opacity_var": 0.0004406541,
    "photosphere_crossed": false
  },
  {
    "step": 3528,
    "epoch": 8,
    "batch_in_epoch": 251,
    "theta1": 0.005092,
    "theta2": 0.007727,
    "loss": 0.046438,
    "eta2d_x": 0.903104,
    "eta2d_y": 0.429423,
    "speed": 3.8e-05,
    "heading_rad": 2.264921,
    "drift_mean": 0.01119037,
    "drift_var": 0.0011787071,
    "opacity_var": 0.0006662537,
    "photosphere_crossed": false
  },
  {
    "step": 3532,
    "epoch": 8,
    "batch_in_epoch": 255,
    "theta1": 0.004942,
    "theta2": 0.007893,
    "loss": 0.009614,
    "eta2d_x": 0.846626,
    "eta2d_y": 0.532188,
    "speed": 4.4e-05,
    "heading_rad": 2.279822,
    "drift_mean": -0.00920589,
    "drift_var": 0.0001483221,
    "opacity_var": 0.0006870427,
    "photosphere_crossed": false
  },
  {
    "step": 3536,
    "epoch": 8,
    "batch_in_epoch": 259,
    "theta1": 0.004863,
    "theta2": 0.007988,
    "loss": 0.006169,
    "eta2d_x": 0.808863,
    "eta2d_y": 0.587997,
    "speed": 2.6e-05,
    "heading_rad": 1.972021,
    "drift_mean": -0.00086137,
    "drift_var": 0.0002661039,
    "opacity_var": 0.0004106152,
    "photosphere_crossed": false
  },
  {
    "step": 3540,
    "epoch": 8,
    "batch_in_epoch": 263,
    "theta1": 0.004886,
    "theta2": 0.008098,
    "loss": 0.016084,
    "eta2d_x": 0.784637,
    "eta2d_y": 0.619955,
    "speed": 3.3e-05,
    "heading_rad": 1.217638,
    "drift_mean": 0.00247889,
    "drift_var": 3.28342e-05,
    "opacity_var": 0.0001354269,
    "photosphere_crossed": true
  },
  {
    "step": 3544,
    "epoch": 8,
    "batch_in_epoch": 267,
    "theta1": 0.0049,
    "theta2": 0.008207,
    "loss": 0.03583,
    "eta2d_x": 0.760153,
    "eta2d_y": 0.649745,
    "speed": 3.7e-05,
    "heading_rad": 2.068132,
    "drift_mean": 0.0049364,
    "drift_var": 0.000105298,
    "opacity_var": 0.0001477759,
    "photosphere_crossed": true
  },
  {
    "step": 3548,
    "epoch": 8,
    "batch_in_epoch": 271,
    "theta1": 0.00486,
    "theta2": 0.008329,
    "loss": 0.011117,
    "eta2d_x": 0.720894,
    "eta2d_y": 0.693045,
    "speed": 3e-05,
    "heading_rad": 1.897383,
    "drift_mean": -0.00617822,
    "drift_var": 0.00010953,
    "opacity_var": 0.0001373862,
    "photosphere_crossed": true
  },
  {
    "step": 3552,
    "epoch": 8,
    "batch_in_epoch": 275,
    "theta1": 0.004853,
    "theta2": 0.008469,
    "loss": 0.048123,
    "eta2d_x": 0.68566,
    "eta2d_y": 0.727922,
    "speed": 4.6e-05,
    "heading_rad": 1.301949,
    "drift_mean": 0.00925153,
    "drift_var": 0.0001017422,
    "opacity_var": 0.0001874242,
    "photosphere_crossed": true
  },
  {
    "step": 3556,
    "epoch": 8,
    "batch_in_epoch": 279,
    "theta1": 0.004833,
    "theta2": 0.008628,
    "loss": 0.012638,
    "eta2d_x": 0.644112,
    "eta2d_y": 0.764931,
    "speed": 3.7e-05,
    "heading_rad": 1.994246,
    "drift_mean": -0.00887114,
    "drift_var": 0.0005249033,
    "opacity_var": 0.0003517305,
    "photosphere_crossed": false
  },
  {
    "step": 3560,
    "epoch": 8,
    "batch_in_epoch": 283,
    "theta1": 0.004754,
    "theta2": 0.008712,
    "loss": 0.011031,
    "eta2d_x": 0.604415,
    "eta2d_y": 0.79667,
    "speed": 3.4e-05,
    "heading_rad": 2.471761,
    "drift_mean": -0.00040198,
    "drift_var": 4.35491e-05,
    "opacity_var": 0.000333211,
    "photosphere_crossed": false
  },
  {
    "step": 3564,
    "epoch": 8,
    "batch_in_epoch": 287,
    "theta1": 0.004606,
    "theta2": 0.008827,
    "loss": 0.010043,
    "eta2d_x": 0.536304,
    "eta2d_y": 0.844025,
    "speed": 4.7e-05,
    "heading_rad": 2.535688,
    "drift_mean": -0.00024689,
    "drift_var": 7.3476e-06,
    "opacity_var": 2.26263e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3568,
    "epoch": 8,
    "batch_in_epoch": 291,
    "theta1": 0.004444,
    "theta2": 0.008837,
    "loss": 0.007351,
    "eta2d_x": 0.4801,
    "eta2d_y": 0.877214,
    "speed": 3.5e-05,
    "heading_rad": -3.058351,
    "drift_mean": -0.00067292,
    "drift_var": 0.0003571294,
    "opacity_var": 0.0001643834,
    "photosphere_crossed": true
  },
  {
    "step": 3572,
    "epoch": 8,
    "batch_in_epoch": 295,
    "theta1": 0.004369,
    "theta2": 0.008774,
    "loss": 0.017024,
    "eta2d_x": 0.465449,
    "eta2d_y": 0.885074,
    "speed": 3.6e-05,
    "heading_rad": -1.938599,
    "drift_mean": 0.00241815,
    "drift_var": 0.000138506,
    "opacity_var": 0.0002257147,
    "photosphere_crossed": true
  },
  {
    "step": 3576,
    "epoch": 8,
    "batch_in_epoch": 299,
    "theta1": 0.004271,
    "theta2": 0.0086,
    "loss": 0.061265,
    "eta2d_x": 0.462676,
    "eta2d_y": 0.886527,
    "speed": 4.1e-05,
    "heading_rad": -2.289854,
    "drift_mean": 0.01106034,
    "drift_var": 0.0001846028,
    "opacity_var": 0.0001664032,
    "photosphere_crossed": true
  },
  {
    "step": 3580,
    "epoch": 8,
    "batch_in_epoch": 303,
    "theta1": 0.004193,
    "theta2": 0.008456,
    "loss": 0.036719,
    "eta2d_x": 0.461502,
    "eta2d_y": 0.887139,
    "speed": 3.6e-05,
    "heading_rad": -2.460929,
    "drift_mean": -0.00613656,
    "drift_var": 0.0008506321,
    "opacity_var": 0.000530431,
    "photosphere_crossed": false
  },
  {
    "step": 3584,
    "epoch": 8,
    "batch_in_epoch": 307,
    "theta1": 0.004174,
    "theta2": 0.008448,
    "loss": 0.031083,
    "eta2d_x": 0.454513,
    "eta2d_y": 0.89074,
    "speed": 1.9e-05,
    "heading_rad": -3.105277,
    "drift_mean": -0.00140904,
    "drift_var": 0.0001388646,
    "opacity_var": 0.0005321231,
    "photosphere_crossed": false
  },
  {
    "step": 3588,
    "epoch": 8,
    "batch_in_epoch": 311,
    "theta1": 0.004067,
    "theta2": 0.008414,
    "loss": 0.02088,
    "eta2d_x": 0.410734,
    "eta2d_y": 0.911755,
    "speed": 3.5e-05,
    "heading_rad": 3.089764,
    "drift_mean": -0.00255078,
    "drift_var": 0.0010018791,
    "opacity_var": 0.0005554376,
    "photosphere_crossed": false
  },
  {
    "step": 3592,
    "epoch": 8,
    "batch_in_epoch": 315,
    "theta1": 0.004012,
    "theta2": 0.008468,
    "loss": 0.019303,
    "eta2d_x": 0.370728,
    "eta2d_y": 0.928741,
    "speed": 1.9e-05,
    "heading_rad": 2.550423,
    "drift_mean": -0.00039421,
    "drift_var": 1.62792e-05,
    "opacity_var": 0.0004784157,
    "photosphere_crossed": false
  },
  {
    "step": 3596,
    "epoch": 8,
    "batch_in_epoch": 319,
    "theta1": 0.003962,
    "theta2": 0.008535,
    "loss": 0.011529,
    "eta2d_x": 0.331614,
    "eta2d_y": 0.943415,
    "speed": 2.4e-05,
    "heading_rad": 1.811345,
    "drift_mean": -0.00194338,
    "drift_var": 0.0001100429,
    "opacity_var": 0.0001521866,
    "photosphere_crossed": true
  },
  {
    "step": 3600,
    "epoch": 8,
    "batch_in_epoch": 323,
    "theta1": 0.003995,
    "theta2": 0.008597,
    "loss": 0.005481,
    "eta2d_x": 0.337005,
    "eta2d_y": 0.941503,
    "speed": 1.3e-05,
    "heading_rad": 0.700363,
    "drift_mean": -0.00151197,
    "drift_var": 1.30846e-05,
    "opacity_var": 5.85037e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3604,
    "epoch": 8,
    "batch_in_epoch": 327,
    "theta1": 0.003922,
    "theta2": 0.008534,
    "loss": 0.012281,
    "eta2d_x": 0.310595,
    "eta2d_y": 0.950542,
    "speed": 3.5e-05,
    "heading_rad": -2.238743,
    "drift_mean": 0.00169998,
    "drift_var": 0.0005510556,
    "opacity_var": 0.0002578833,
    "photosphere_crossed": true
  },
  {
    "step": 3608,
    "epoch": 8,
    "batch_in_epoch": 331,
    "theta1": 0.003901,
    "theta2": 0.008459,
    "loss": 0.056829,
    "eta2d_x": 0.312911,
    "eta2d_y": 0.949782,
    "speed": 1.4e-05,
    "heading_rad": -1.434956,
    "drift_mean": 0.01113691,
    "drift_var": 0.0005877598,
    "opacity_var": 0.0005386145,
    "photosphere_crossed": false
  },
  {
    "step": 3612,
    "epoch": 8,
    "batch_in_epoch": 335,
    "theta1": 0.003973,
    "theta2": 0.008349,
    "loss": 0.021111,
    "eta2d_x": 0.375358,
    "eta2d_y": 0.92688,
    "speed": 4.2e-05,
    "heading_rad": -0.80773,
    "drift_mean": -0.00892945,
    "drift_var": 0.0001987129,
    "opacity_var": 0.0004920499,
    "photosphere_crossed": false
  },
  {
    "step": 3616,
    "epoch": 8,
    "batch_in_epoch": 339,
    "theta1": 0.004018,
    "theta2": 0.00815,
    "loss": 0.010002,
    "eta2d_x": 0.452381,
    "eta2d_y": 0.891825,
    "speed": 4e-05,
    "heading_rad": -1.198117,
    "drift_mean": -0.00277726,
    "drift_var": 0.0001010822,
    "opacity_var": 0.0004051487,
    "photosphere_crossed": false
  },
  {
    "step": 3620,
    "epoch": 8,
    "batch_in_epoch": 343,
    "theta1": 0.004132,
    "theta2": 0.008067,
    "loss": 0.006722,
    "eta2d_x": 0.54093,
    "eta2d_y": 0.841068,
    "speed": 3.6e-05,
    "heading_rad": -0.597407,
    "drift_mean": -0.00082017,
    "drift_var": 0.0011044608,
    "opacity_var": 0.0005369053,
    "photosphere_crossed": false
  },
  {
    "step": 3624,
    "epoch": 8,
    "batch_in_epoch": 347,
    "theta1": 0.004179,
    "theta2": 0.007985,
    "loss": 0.039545,
    "eta2d_x": 0.593096,
    "eta2d_y": 0.805132,
    "speed": 2.9e-05,
    "heading_rad": -1.388599,
    "drift_mean": 0.00820596,
    "drift_var": 0.0006601612,
    "opacity_var": 0.000803394,
    "photosphere_crossed": false
  },
  {
    "step": 3628,
    "epoch": 8,
    "batch_in_epoch": 351,
    "theta1": 0.004141,
    "theta2": 0.007871,
    "loss": 0.027511,
    "eta2d_x": 0.617382,
    "eta2d_y": 0.786663,
    "speed": 1.3e-05,
    "heading_rad": -2.988442,
    "drift_mean": -0.00300856,
    "drift_var": 0.0002388935,
    "opacity_var": 0.0006821621,
    "photosphere_crossed": false
  },
  {
    "step": 3632,
    "epoch": 8,
    "batch_in_epoch": 355,
    "theta1": 0.004122,
    "theta2": 0.007898,
    "loss": 0.007635,
    "eta2d_x": 0.596899,
    "eta2d_y": 0.802317,
    "speed": 1e-05,
    "heading_rad": 1.826898,
    "drift_mean": -0.00496902,
    "drift_var": 2.1702e-05,
    "opacity_var": 0.0001225962,
    "photosphere_crossed": true
  },
  {
    "step": 3636,
    "epoch": 8,
    "batch_in_epoch": 359,
    "theta1": 0.004111,
    "theta2": 0.007988,
    "loss": 0.003103,
    "eta2d_x": 0.55712,
    "eta2d_y": 0.830432,
    "speed": 2.7e-05,
    "heading_rad": 1.57396,
    "drift_mean": -0.001133,
    "drift_var": 0.0014086477,
    "opacity_var": 0.0006747502,
    "photosphere_crossed": false
  },
  {
    "step": 3640,
    "epoch": 8,
    "batch_in_epoch": 363,
    "theta1": 0.00403,
    "theta2": 0.007976,
    "loss": 0.030465,
    "eta2d_x": 0.515855,
    "eta2d_y": 0.856676,
    "speed": 3.1e-05,
    "heading_rad": -2.362287,
    "drift_mean": 0.00684038,
    "drift_var": 0.0002570524,
    "opacity_var": 0.0007581566,
    "photosphere_crossed": false
  },
  {
    "step": 3644,
    "epoch": 8,
    "batch_in_epoch": 367,
    "theta1": 0.003897,
    "theta2": 0.007905,
    "loss": 0.013827,
    "eta2d_x": 0.455017,
    "eta2d_y": 0.890483,
    "speed": 4.1e-05,
    "heading_rad": -2.813147,
    "drift_mean": -0.00415944,
    "drift_var": 0.0004016533,
    "opacity_var": 0.0006831146,
    "photosphere_crossed": false
  },
  {
    "step": 3648,
    "epoch": 8,
    "batch_in_epoch": 371,
    "theta1": 0.003906,
    "theta2": 0.00799,
    "loss": 0.034495,
    "eta2d_x": 0.431727,
    "eta2d_y": 0.902004,
    "speed": 5e-06,
    "heading_rad": 0.591913,
    "drift_mean": 0.00516712,
    "drift_var": 0.0001645266,
    "opacity_var": 0.0002747726,
    "photosphere_crossed": true
  },
  {
    "step": 3652,
    "epoch": 8,
    "batch_in_epoch": 375,
    "theta1": 0.003949,
    "theta2": 0.008023,
    "loss": 0.007719,
    "eta2d_x": 0.449467,
    "eta2d_y": 0.893297,
    "speed": 2.3e-05,
    "heading_rad": 0.789278,
    "drift_mean": -0.006694,
    "drift_var": 0.0001637301,
    "opacity_var": 0.0001773551,
    "photosphere_crossed": true
  },
  {
    "step": 3656,
    "epoch": 8,
    "batch_in_epoch": 379,
    "theta1": 0.003988,
    "theta2": 0.008095,
    "loss": 0.007439,
    "eta2d_x": 0.451329,
    "eta2d_y": 0.892358,
    "speed": 1e-05,
    "heading_rad": 1.176833,
    "drift_mean": -7.009e-05,
    "drift_var": 4.46578e-05,
    "opacity_var": 0.0001127998,
    "photosphere_crossed": true
  },
  {
    "step": 3660,
    "epoch": 8,
    "batch_in_epoch": 383,
    "theta1": 0.004,
    "theta2": 0.008099,
    "loss": 0.020629,
    "eta2d_x": 0.457421,
    "eta2d_y": 0.88925,
    "speed": 8e-06,
    "heading_rad": -1.047936,
    "drift_mean": 0.00329752,
    "drift_var": 8.92756e-05,
    "opacity_var": 7.11048e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3664,
    "epoch": 8,
    "batch_in_epoch": 387,
    "theta1": 0.00403,
    "theta2": 0.008061,
    "loss": 0.006937,
    "eta2d_x": 0.48681,
    "eta2d_y": 0.873508,
    "speed": 1.6e-05,
    "heading_rad": -1.081692,
    "drift_mean": -0.00342307,
    "drift_var": 6.68835e-05,
    "opacity_var": 8.81635e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3668,
    "epoch": 8,
    "batch_in_epoch": 391,
    "theta1": 0.004039,
    "theta2": 0.007999,
    "loss": 0.029407,
    "eta2d_x": 0.512955,
    "eta2d_y": 0.858416,
    "speed": 1.8e-05,
    "heading_rad": -1.778325,
    "drift_mean": 0.00561768,
    "drift_var": 0.0011109009,
    "opacity_var": 0.0005635934,
    "photosphere_crossed": false
  },
  {
    "step": 3672,
    "epoch": 8,
    "batch_in_epoch": 395,
    "theta1": 0.004055,
    "theta2": 0.007982,
    "loss": 0.021076,
    "eta2d_x": 0.528467,
    "eta2d_y": 0.848954,
    "speed": 1e-05,
    "heading_rad": -1.039968,
    "drift_mean": -0.00208286,
    "drift_var": 0.001257682,
    "opacity_var": 0.0010670334,
    "photosphere_crossed": false
  },
  {
    "step": 3676,
    "epoch": 8,
    "batch_in_epoch": 399,
    "theta1": 0.004073,
    "theta2": 0.007947,
    "loss": 0.059584,
    "eta2d_x": 0.551733,
    "eta2d_y": 0.834021,
    "speed": 1.1e-05,
    "heading_rad": -1.128817,
    "drift_mean": 0.00962712,
    "drift_var": 0.0002974487,
    "opacity_var": 0.0007482612,
    "photosphere_crossed": false
  },
  {
    "step": 3680,
    "epoch": 8,
    "batch_in_epoch": 403,
    "theta1": 0.004077,
    "theta2": 0.007856,
    "loss": 0.016373,
    "eta2d_x": 0.589164,
    "eta2d_y": 0.808013,
    "speed": 2.9e-05,
    "heading_rad": -1.680563,
    "drift_mean": -0.01080293,
    "drift_var": 0.0004999405,
    "opacity_var": 0.000466965,
    "photosphere_crossed": false
  },
  {
    "step": 3684,
    "epoch": 8,
    "batch_in_epoch": 407,
    "theta1": 0.00406,
    "theta2": 0.007746,
    "loss": 0.029815,
    "eta2d_x": 0.627313,
    "eta2d_y": 0.778767,
    "speed": 2.4e-05,
    "heading_rad": -1.130337,
    "drift_mean": 0.00336068,
    "drift_var": 9.90641e-05,
    "opacity_var": 0.0004871806,
    "photosphere_crossed": false
  },
  {
    "step": 3688,
    "epoch": 8,
    "batch_in_epoch": 411,
    "theta1": 0.004135,
    "theta2": 0.00772,
    "loss": 0.065672,
    "eta2d_x": 0.679716,
    "eta2d_y": 0.733476,
    "speed": 2.8e-05,
    "heading_rad": -0.156757,
    "drift_mean": 0.00896409,
    "drift_var": 0.0007347023,
    "opacity_var": 0.000429725,
    "photosphere_crossed": false
  },
  {
    "step": 3692,
    "epoch": 8,
    "batch_in_epoch": 415,
    "theta1": 0.004207,
    "theta2": 0.007733,
    "loss": 0.033011,
    "eta2d_x": 0.706739,
    "eta2d_y": 0.707474,
    "speed": 1.2e-05,
    "heading_rad": 0.716534,
    "drift_mean": -0.00816523,
    "drift_var": 0.001222178,
    "opacity_var": 0.0009467565,
    "photosphere_crossed": false
  },
  {
    "step": 3696,
    "epoch": 8,
    "batch_in_epoch": 419,
    "theta1": 0.004249,
    "theta2": 0.007817,
    "loss": 0.031812,
    "eta2d_x": 0.68981,
    "eta2d_y": 0.723991,
    "speed": 3.6e-05,
    "heading_rad": 0.96451,
    "drift_mean": -0.00029973,
    "drift_var": 0.0001527117,
    "opacity_var": 0.0009581002,
    "photosphere_crossed": false
  },
  {
    "step": 3700,
    "epoch": 8,
    "batch_in_epoch": 423,
    "theta1": 0.004395,
    "theta2": 0.007818,
    "loss": 0.018807,
    "eta2d_x": 0.744094,
    "eta2d_y": 0.668075,
    "speed": 3.4e-05,
    "heading_rad": -0.130615,
    "drift_mean": -0.0032513,
    "drift_var": 0.0001971867,
    "opacity_var": 0.0002132634,
    "photosphere_crossed": true
  },
  {
    "step": 3704,
    "epoch": 8,
    "batch_in_epoch": 427,
    "theta1": 0.004496,
    "theta2": 0.007854,
    "loss": 0.013813,
    "eta2d_x": 0.762908,
    "eta2d_y": 0.646507,
    "speed": 2.9e-05,
    "heading_rad": 0.733002,
    "drift_mean": -0.0012485,
    "drift_var": 4.20837e-05,
    "opacity_var": 0.0001201399,
    "photosphere_crossed": true
  },
  {
    "step": 3708,
    "epoch": 8,
    "batch_in_epoch": 431,
    "theta1": 0.004519,
    "theta2": 0.007887,
    "loss": 0.024808,
    "eta2d_x": 0.758184,
    "eta2d_y": 0.65204,
    "speed": 1.1e-05,
    "heading_rad": 0.636895,
    "drift_mean": 0.00274896,
    "drift_var": 2.02539e-05,
    "opacity_var": 5.81915e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3712,
    "epoch": 8,
    "batch_in_epoch": 435,
    "theta1": 0.004518,
    "theta2": 0.007865,
    "loss": 0.016806,
    "eta2d_x": 0.765424,
    "eta2d_y": 0.643527,
    "speed": 1.4e-05,
    "heading_rad": -3.034213,
    "drift_mean": -0.00200074,
    "drift_var": 0.0016618665,
    "opacity_var": 0.0007590524,
    "photosphere_crossed": false
  },
  {
    "step": 3716,
    "epoch": 8,
    "batch_in_epoch": 439,
    "theta1": 0.004437,
    "theta2": 0.007818,
    "loss": 0.027557,
    "eta2d_x": 0.757679,
    "eta2d_y": 0.652627,
    "speed": 3.4e-05,
    "heading_rad": -2.237947,
    "drift_mean": 0.00268795,
    "drift_var": 8.21444e-05,
    "opacity_var": 0.0007895191,
    "photosphere_crossed": false
  },
  {
    "step": 3720,
    "epoch": 8,
    "batch_in_epoch": 443,
    "theta1": 0.004389,
    "theta2": 0.007731,
    "loss": 0.008796,
    "eta2d_x": 0.775008,
    "eta2d_y": 0.631952,
    "speed": 2.3e-05,
    "heading_rad": -2.015035,
    "drift_mean": -0.00469029,
    "drift_var": 3.34856e-05,
    "opacity_var": 6.41672e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3724,
    "epoch": 8,
    "batch_in_epoch": 447,
    "theta1": 0.004359,
    "theta2": 0.007704,
    "loss": 0.013377,
    "eta2d_x": 0.776188,
    "eta2d_y": 0.630501,
    "speed": 1.5e-05,
    "heading_rad": 3.136238,
    "drift_mean": 0.00114521,
    "drift_var": 8.1491e-05,
    "opacity_var": 8.92867e-05,
    "photosphere_crossed": true
  },
  {
    "step": 3728,
    "epoch": 8,
    "batch_in_epoch": 451,
    "theta1": 0.004354,
    "theta2": 0.007697,
    "loss": 0.045815,
    "eta2d_x": 0.77746,
    "eta2d_y": 0.628933,
    "speed": 1.1e-05,
    "heading_rad": 2.377468,
    "drift_mean": 0.00810959,
    "drift_var": 0.0008754085,
    "opacity_var": 0.0004364078,
    "photosphere_crossed": false
  },
  {
    "step": 3732,
    "epoch": 8,
    "batch_in_epoch": 455,
    "theta1": 0.004341,
    "theta2": 0.007711,
    "loss": 0.004134,
    "eta2d_x": 0.767638,
    "eta2d_y": 0.640884,
    "speed": 1e-06,
    "heading_rad": 2.201507,
    "drift_mean": -0.01042041,
    "drift_var": 0.000288033,
    "opacity_var": 0.0005935022,
    "photosphere_crossed": false
  },
  {
    "step": 3736,
    "epoch": 8,
    "batch_in_epoch": 459,
    "theta1": 0.004288,
    "theta2": 0.007709,
    "loss": 0.005723,
    "eta2d_x": 0.749746,
    "eta2d_y": 0.661725,
    "speed": 7e-06,
    "heading_rad": 3.070432,
    "drift_mean": 0.00039734,
    "drift_var": 0.0003083849,
    "opacity_var": 0.0004660075,
    "photosphere_crossed": false
  },
  {
    "step": 3740,
    "epoch": 8,
    "batch_in_epoch": 463,
    "theta1": 0.004237,
    "theta2": 0.007648,
    "loss": 0.022599,
    "eta2d_x": 0.756756,
    "eta2d_y": 0.653697,
    "speed": 3.2e-05,
    "heading_rad": -1.871772,
    "drift_mean": 0.00421886,
    "drift_var": 0.0006893278,
    "opacity_var": 0.0004827745,
    "photosphere_crossed": false
  },
  {
    "step": 3744,
    "epoch": 8,
    "batch_in_epoch": 467,
    "theta1": 0.004233,
    "theta2": 0.007575,
    "loss": 0.024979,
    "eta2d_x": 0.788058,
    "eta2d_y": 0.615601,
    "speed": 2e-05,
    "heading_rad": -0.333137,
    "drift_mean": 0.00059508,
    "drift_var": 0.0001558932,
    "opacity_var": 0.0003797485,
    "photosphere_crossed": false
  }
]