{
  "best_metric": 0.7727344036102295,
  "best_model_checkpoint": "ckpt/origin/vehicle_environment/checkpoint-91",
  "epoch": 4.830917874396135,
  "eval_steps": 7,
  "global_step": 125,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.04,
      "learning_rate": 0.0,
      "loss": 0.8677,
      "step": 1
    },
    {
      "epoch": 0.08,
      "learning_rate": 1e-05,
      "loss": 0.8182,
      "step": 2
    },
    {
      "epoch": 0.12,
      "learning_rate": 1e-05,
      "loss": 0.8928,
      "step": 3
    },
    {
      "epoch": 0.15,
      "learning_rate": 1e-05,
      "loss": 0.9758,
      "step": 4
    },
    {
      "epoch": 0.19,
      "learning_rate": 1e-05,
      "loss": 0.8773,
      "step": 5
    },
    {
      "epoch": 0.23,
      "learning_rate": 1e-05,
      "loss": 0.8984,
      "step": 6
    },
    {
      "epoch": 0.27,
      "learning_rate": 1e-05,
      "loss": 0.793,
      "step": 7
    },
    {
      "epoch": 0.27,
      "eval_loss": 0.954296886920929,
      "eval_runtime": 44.1101,
      "eval_samples_per_second": 1.134,
      "eval_steps_per_second": 0.567,
      "step": 7
    },
    {
      "epoch": 0.31,
      "learning_rate": 1e-05,
      "loss": 0.8718,
      "step": 8
    },
    {
      "epoch": 0.35,
      "learning_rate": 1e-05,
      "loss": 0.8806,
      "step": 9
    },
    {
      "epoch": 0.39,
      "learning_rate": 1e-05,
      "loss": 0.7152,
      "step": 10
    },
    {
      "epoch": 0.43,
      "learning_rate": 1e-05,
      "loss": 0.7653,
      "step": 11
    },
    {
      "epoch": 0.46,
      "learning_rate": 1e-05,
      "loss": 0.7864,
      "step": 12
    },
    {
      "epoch": 0.5,
      "learning_rate": 1e-05,
      "loss": 0.7629,
      "step": 13
    },
    {
      "epoch": 0.54,
      "learning_rate": 1e-05,
      "loss": 0.8066,
      "step": 14
    },
    {
      "epoch": 0.54,
      "eval_loss": 0.8725000023841858,
      "eval_runtime": 43.3056,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 14
    },
    {
      "epoch": 0.58,
      "learning_rate": 1e-05,
      "loss": 0.793,
      "step": 15
    },
    {
      "epoch": 0.62,
      "learning_rate": 1e-05,
      "loss": 0.7722,
      "step": 16
    },
    {
      "epoch": 0.66,
      "learning_rate": 1e-05,
      "loss": 0.8317,
      "step": 17
    },
    {
      "epoch": 0.7,
      "learning_rate": 1e-05,
      "loss": 0.7793,
      "step": 18
    },
    {
      "epoch": 0.73,
      "learning_rate": 1e-05,
      "loss": 0.6925,
      "step": 19
    },
    {
      "epoch": 0.77,
      "learning_rate": 1e-05,
      "loss": 0.7365,
      "step": 20
    },
    {
      "epoch": 0.81,
      "learning_rate": 1e-05,
      "loss": 0.7505,
      "step": 21
    },
    {
      "epoch": 0.81,
      "eval_loss": 0.8274219036102295,
      "eval_runtime": 43.2778,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.578,
      "step": 21
    },
    {
      "epoch": 0.85,
      "learning_rate": 1e-05,
      "loss": 0.7391,
      "step": 22
    },
    {
      "epoch": 0.89,
      "learning_rate": 1e-05,
      "loss": 0.7058,
      "step": 23
    },
    {
      "epoch": 0.93,
      "learning_rate": 1e-05,
      "loss": 0.6859,
      "step": 24
    },
    {
      "epoch": 0.97,
      "learning_rate": 1e-05,
      "loss": 0.679,
      "step": 25
    },
    {
      "epoch": 1.0,
      "learning_rate": 1e-05,
      "loss": 0.6708,
      "step": 26
    },
    {
      "epoch": 1.04,
      "learning_rate": 1e-05,
      "loss": 0.6968,
      "step": 27
    },
    {
      "epoch": 1.08,
      "learning_rate": 1e-05,
      "loss": 0.6937,
      "step": 28
    },
    {
      "epoch": 1.08,
      "eval_loss": 0.802539050579071,
      "eval_runtime": 43.6,
      "eval_samples_per_second": 1.147,
      "eval_steps_per_second": 0.573,
      "step": 28
    },
    {
      "epoch": 1.12,
      "learning_rate": 1e-05,
      "loss": 0.7578,
      "step": 29
    },
    {
      "epoch": 1.16,
      "learning_rate": 1e-05,
      "loss": 0.6307,
      "step": 30
    },
    {
      "epoch": 1.2,
      "learning_rate": 1e-05,
      "loss": 0.6949,
      "step": 31
    },
    {
      "epoch": 1.24,
      "learning_rate": 1e-05,
      "loss": 0.6743,
      "step": 32
    },
    {
      "epoch": 1.28,
      "learning_rate": 1e-05,
      "loss": 0.6814,
      "step": 33
    },
    {
      "epoch": 1.31,
      "learning_rate": 1e-05,
      "loss": 0.5865,
      "step": 34
    },
    {
      "epoch": 1.35,
      "learning_rate": 1e-05,
      "loss": 0.5945,
      "step": 35
    },
    {
      "epoch": 1.35,
      "eval_loss": 0.7889453172683716,
      "eval_runtime": 43.2969,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 35
    },
    {
      "epoch": 1.39,
      "learning_rate": 1e-05,
      "loss": 0.6505,
      "step": 36
    },
    {
      "epoch": 1.43,
      "learning_rate": 1e-05,
      "loss": 0.7418,
      "step": 37
    },
    {
      "epoch": 1.47,
      "learning_rate": 1e-05,
      "loss": 0.6592,
      "step": 38
    },
    {
      "epoch": 1.51,
      "learning_rate": 1e-05,
      "loss": 0.6829,
      "step": 39
    },
    {
      "epoch": 1.55,
      "learning_rate": 1e-05,
      "loss": 0.7168,
      "step": 40
    },
    {
      "epoch": 1.58,
      "learning_rate": 1e-05,
      "loss": 0.7255,
      "step": 41
    },
    {
      "epoch": 1.62,
      "learning_rate": 1e-05,
      "loss": 0.7029,
      "step": 42
    },
    {
      "epoch": 1.62,
      "eval_loss": 0.7841406464576721,
      "eval_runtime": 43.3346,
      "eval_samples_per_second": 1.154,
      "eval_steps_per_second": 0.577,
      "step": 42
    },
    {
      "epoch": 1.66,
      "learning_rate": 1e-05,
      "loss": 0.589,
      "step": 43
    },
    {
      "epoch": 1.7,
      "learning_rate": 1e-05,
      "loss": 0.6757,
      "step": 44
    },
    {
      "epoch": 1.74,
      "learning_rate": 1e-05,
      "loss": 0.7625,
      "step": 45
    },
    {
      "epoch": 1.78,
      "learning_rate": 1e-05,
      "loss": 0.6877,
      "step": 46
    },
    {
      "epoch": 1.82,
      "learning_rate": 1e-05,
      "loss": 0.6104,
      "step": 47
    },
    {
      "epoch": 1.86,
      "learning_rate": 1e-05,
      "loss": 0.6581,
      "step": 48
    },
    {
      "epoch": 1.89,
      "learning_rate": 1e-05,
      "loss": 0.6228,
      "step": 49
    },
    {
      "epoch": 1.89,
      "eval_loss": 0.7757421731948853,
      "eval_runtime": 43.3021,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 49
    },
    {
      "epoch": 1.93,
      "learning_rate": 1e-05,
      "loss": 0.7766,
      "step": 50
    },
    {
      "epoch": 1.97,
      "learning_rate": 1e-05,
      "loss": 0.5925,
      "step": 51
    },
    {
      "epoch": 2.01,
      "learning_rate": 1e-05,
      "loss": 0.6484,
      "step": 52
    },
    {
      "epoch": 2.05,
      "learning_rate": 1e-05,
      "loss": 0.7354,
      "step": 53
    },
    {
      "epoch": 2.09,
      "learning_rate": 1e-05,
      "loss": 0.6472,
      "step": 54
    },
    {
      "epoch": 2.13,
      "learning_rate": 1e-05,
      "loss": 0.64,
      "step": 55
    },
    {
      "epoch": 2.16,
      "learning_rate": 1e-05,
      "loss": 0.6172,
      "step": 56
    },
    {
      "epoch": 2.16,
      "eval_loss": 0.7679296731948853,
      "eval_runtime": 43.362,
      "eval_samples_per_second": 1.153,
      "eval_steps_per_second": 0.577,
      "step": 56
    },
    {
      "epoch": 2.2,
      "learning_rate": 1e-05,
      "loss": 0.6298,
      "step": 57
    },
    {
      "epoch": 2.24,
      "learning_rate": 1e-05,
      "loss": 0.5544,
      "step": 58
    },
    {
      "epoch": 2.28,
      "learning_rate": 1e-05,
      "loss": 0.6791,
      "step": 59
    },
    {
      "epoch": 2.32,
      "learning_rate": 1e-05,
      "loss": 0.6732,
      "step": 60
    },
    {
      "epoch": 2.36,
      "learning_rate": 1e-05,
      "loss": 0.6169,
      "step": 61
    },
    {
      "epoch": 2.4,
      "learning_rate": 1e-05,
      "loss": 0.6215,
      "step": 62
    },
    {
      "epoch": 2.43,
      "learning_rate": 1e-05,
      "loss": 0.5446,
      "step": 63
    },
    {
      "epoch": 2.43,
      "eval_loss": 0.7676171660423279,
      "eval_runtime": 43.2939,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 63
    },
    {
      "epoch": 2.47,
      "learning_rate": 1e-05,
      "loss": 0.5829,
      "step": 64
    },
    {
      "epoch": 2.51,
      "learning_rate": 1e-05,
      "loss": 0.6256,
      "step": 65
    },
    {
      "epoch": 2.55,
      "learning_rate": 1e-05,
      "loss": 0.5826,
      "step": 66
    },
    {
      "epoch": 2.59,
      "learning_rate": 1e-05,
      "loss": 0.6113,
      "step": 67
    },
    {
      "epoch": 2.63,
      "learning_rate": 1e-05,
      "loss": 0.6292,
      "step": 68
    },
    {
      "epoch": 2.67,
      "learning_rate": 1e-05,
      "loss": 0.6434,
      "step": 69
    },
    {
      "epoch": 2.71,
      "learning_rate": 1e-05,
      "loss": 0.5857,
      "step": 70
    },
    {
      "epoch": 2.71,
      "eval_loss": 0.7694922089576721,
      "eval_runtime": 43.2962,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 70
    },
    {
      "epoch": 2.74,
      "learning_rate": 1e-05,
      "loss": 0.5844,
      "step": 71
    },
    {
      "epoch": 2.78,
      "learning_rate": 1e-05,
      "loss": 0.63,
      "step": 72
    },
    {
      "epoch": 2.82,
      "learning_rate": 1e-05,
      "loss": 0.5413,
      "step": 73
    },
    {
      "epoch": 2.86,
      "learning_rate": 1e-05,
      "loss": 0.6852,
      "step": 74
    },
    {
      "epoch": 2.9,
      "learning_rate": 1e-05,
      "loss": 0.5592,
      "step": 75
    },
    {
      "epoch": 2.94,
      "learning_rate": 1e-05,
      "loss": 0.6597,
      "step": 76
    },
    {
      "epoch": 2.98,
      "learning_rate": 1e-05,
      "loss": 0.5186,
      "step": 77
    },
    {
      "epoch": 2.98,
      "eval_loss": 0.7741405963897705,
      "eval_runtime": 43.2995,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 77
    },
    {
      "epoch": 3.01,
      "learning_rate": 1e-05,
      "loss": 0.6163,
      "step": 78
    },
    {
      "epoch": 3.05,
      "learning_rate": 1e-05,
      "loss": 0.5931,
      "step": 79
    },
    {
      "epoch": 3.09,
      "learning_rate": 1e-05,
      "loss": 0.6509,
      "step": 80
    },
    {
      "epoch": 3.13,
      "learning_rate": 1e-05,
      "loss": 0.6136,
      "step": 81
    },
    {
      "epoch": 3.17,
      "learning_rate": 1e-05,
      "loss": 0.5426,
      "step": 82
    },
    {
      "epoch": 3.21,
      "learning_rate": 1e-05,
      "loss": 0.575,
      "step": 83
    },
    {
      "epoch": 3.25,
      "learning_rate": 1e-05,
      "loss": 0.5386,
      "step": 84
    },
    {
      "epoch": 3.25,
      "eval_loss": 0.7715234160423279,
      "eval_runtime": 43.3518,
      "eval_samples_per_second": 1.153,
      "eval_steps_per_second": 0.577,
      "step": 84
    },
    {
      "epoch": 3.29,
      "learning_rate": 1e-05,
      "loss": 0.5588,
      "step": 85
    },
    {
      "epoch": 3.32,
      "learning_rate": 1e-05,
      "loss": 0.5428,
      "step": 86
    },
    {
      "epoch": 3.36,
      "learning_rate": 1e-05,
      "loss": 0.6062,
      "step": 87
    },
    {
      "epoch": 3.4,
      "learning_rate": 1e-05,
      "loss": 0.5421,
      "step": 88
    },
    {
      "epoch": 3.44,
      "learning_rate": 1e-05,
      "loss": 0.6211,
      "step": 89
    },
    {
      "epoch": 3.48,
      "learning_rate": 1e-05,
      "loss": 0.6537,
      "step": 90
    },
    {
      "epoch": 3.52,
      "learning_rate": 1e-05,
      "loss": 0.5364,
      "step": 91
    },
    {
      "epoch": 3.52,
      "eval_loss": 0.7727344036102295,
      "eval_runtime": 43.2938,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 91
    },
    {
      "epoch": 3.56,
      "learning_rate": 1e-05,
      "loss": 0.5734,
      "step": 92
    },
    {
      "epoch": 3.59,
      "learning_rate": 1e-05,
      "loss": 0.5648,
      "step": 93
    },
    {
      "epoch": 3.63,
      "learning_rate": 1e-05,
      "loss": 0.5326,
      "step": 94
    },
    {
      "epoch": 3.67,
      "learning_rate": 1e-05,
      "loss": 0.5787,
      "step": 95
    },
    {
      "epoch": 3.71,
      "learning_rate": 1e-05,
      "loss": 0.5831,
      "step": 96
    },
    {
      "epoch": 3.75,
      "learning_rate": 1e-05,
      "loss": 0.6093,
      "step": 97
    },
    {
      "epoch": 3.79,
      "learning_rate": 1e-05,
      "loss": 0.6548,
      "step": 98
    },
    {
      "epoch": 3.79,
      "eval_loss": 0.7698047161102295,
      "eval_runtime": 43.3828,
      "eval_samples_per_second": 1.153,
      "eval_steps_per_second": 0.576,
      "step": 98
    },
    {
      "epoch": 3.83,
      "learning_rate": 1e-05,
      "loss": 0.5649,
      "step": 99
    },
    {
      "epoch": 3.86,
      "learning_rate": 1e-05,
      "loss": 0.5951,
      "step": 100
    },
    {
      "epoch": 3.9,
      "learning_rate": 1e-05,
      "loss": 0.5956,
      "step": 101
    },
    {
      "epoch": 3.94,
      "learning_rate": 1e-05,
      "loss": 0.5413,
      "step": 102
    },
    {
      "epoch": 3.98,
      "learning_rate": 1e-05,
      "loss": 0.5199,
      "step": 103
    },
    {
      "epoch": 4.02,
      "learning_rate": 1e-05,
      "loss": 0.5608,
      "step": 104
    },
    {
      "epoch": 4.06,
      "learning_rate": 1e-05,
      "loss": 0.5592,
      "step": 105
    },
    {
      "epoch": 4.06,
      "eval_loss": 0.7690625190734863,
      "eval_runtime": 43.3059,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.577,
      "step": 105
    },
    {
      "epoch": 4.1,
      "learning_rate": 1e-05,
      "loss": 0.5203,
      "step": 106
    },
    {
      "epoch": 4.14,
      "learning_rate": 1e-05,
      "loss": 0.5927,
      "step": 107
    },
    {
      "epoch": 4.17,
      "learning_rate": 1e-05,
      "loss": 0.6014,
      "step": 108
    },
    {
      "epoch": 4.21,
      "learning_rate": 1e-05,
      "loss": 0.5519,
      "step": 109
    },
    {
      "epoch": 4.25,
      "learning_rate": 1e-05,
      "loss": 0.5568,
      "step": 110
    },
    {
      "epoch": 4.29,
      "learning_rate": 1e-05,
      "loss": 0.5682,
      "step": 111
    },
    {
      "epoch": 4.33,
      "learning_rate": 1e-05,
      "loss": 0.5782,
      "step": 112
    },
    {
      "epoch": 4.33,
      "eval_loss": 0.7662500143051147,
      "eval_runtime": 43.2755,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.578,
      "step": 112
    },
    {
      "epoch": 4.37,
      "learning_rate": 1e-05,
      "loss": 0.4916,
      "step": 113
    },
    {
      "epoch": 4.41,
      "learning_rate": 1e-05,
      "loss": 0.5217,
      "step": 114
    },
    {
      "epoch": 4.44,
      "learning_rate": 1e-05,
      "loss": 0.6367,
      "step": 115
    },
    {
      "epoch": 4.48,
      "learning_rate": 1e-05,
      "loss": 0.562,
      "step": 116
    },
    {
      "epoch": 4.52,
      "learning_rate": 1e-05,
      "loss": 0.5706,
      "step": 117
    },
    {
      "epoch": 4.56,
      "learning_rate": 1e-05,
      "loss": 0.5658,
      "step": 118
    },
    {
      "epoch": 4.6,
      "learning_rate": 1e-05,
      "loss": 0.549,
      "step": 119
    },
    {
      "epoch": 4.6,
      "eval_loss": 0.7637890577316284,
      "eval_runtime": 43.2782,
      "eval_samples_per_second": 1.155,
      "eval_steps_per_second": 0.578,
      "step": 119
    },
    {
      "epoch": 4.64,
      "learning_rate": 1e-05,
      "loss": 0.5286,
      "step": 120
    },
    {
      "epoch": 4.68,
      "learning_rate": 1e-05,
      "loss": 0.4961,
      "step": 121
    },
    {
      "epoch": 4.71,
      "learning_rate": 1e-05,
      "loss": 0.5203,
      "step": 122
    },
    {
      "epoch": 4.75,
      "learning_rate": 1e-05,
      "loss": 0.5485,
      "step": 123
    },
    {
      "epoch": 4.79,
      "learning_rate": 1e-05,
      "loss": 0.599,
      "step": 124
    },
    {
      "epoch": 4.83,
      "learning_rate": 1e-05,
      "loss": 0.4822,
      "step": 125
    },
    {
      "epoch": 4.83,
      "step": 125,
      "total_flos": 66871008165888.0,
      "train_loss": 0.6458125,
      "train_runtime": 5591.1555,
      "train_samples_per_second": 1.11,
      "train_steps_per_second": 0.022
    }
  ],
  "logging_steps": 1.0,
  "max_steps": 125,
  "num_train_epochs": 5,
  "save_steps": 13,
  "total_flos": 66871008165888.0,
  "trial_name": null,
  "trial_params": null
}