pythia-exp / trainer_state.json
TinyPixel's picture
Upload folder using huggingface_hub
a629300
raw
history blame
5.66 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9944064636420137,
"eval_steps": 500,
"global_step": 100,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.02,
"learning_rate": 0.0002,
"loss": 2.6239,
"step": 2
},
{
"epoch": 0.04,
"learning_rate": 0.0002,
"loss": 2.5308,
"step": 4
},
{
"epoch": 0.06,
"learning_rate": 0.0002,
"loss": 2.5426,
"step": 6
},
{
"epoch": 0.08,
"learning_rate": 0.0002,
"loss": 2.4938,
"step": 8
},
{
"epoch": 0.1,
"learning_rate": 0.0002,
"loss": 2.3458,
"step": 10
},
{
"epoch": 0.12,
"learning_rate": 0.0002,
"loss": 2.3949,
"step": 12
},
{
"epoch": 0.14,
"learning_rate": 0.0002,
"loss": 2.4183,
"step": 14
},
{
"epoch": 0.16,
"learning_rate": 0.0002,
"loss": 2.3387,
"step": 16
},
{
"epoch": 0.18,
"learning_rate": 0.0002,
"loss": 2.2117,
"step": 18
},
{
"epoch": 0.2,
"learning_rate": 0.0002,
"loss": 2.0359,
"step": 20
},
{
"epoch": 0.22,
"learning_rate": 0.0002,
"loss": 2.1539,
"step": 22
},
{
"epoch": 0.24,
"learning_rate": 0.0002,
"loss": 2.3478,
"step": 24
},
{
"epoch": 0.26,
"learning_rate": 0.0002,
"loss": 2.1061,
"step": 26
},
{
"epoch": 0.28,
"learning_rate": 0.0002,
"loss": 2.2001,
"step": 28
},
{
"epoch": 0.3,
"learning_rate": 0.0002,
"loss": 2.2459,
"step": 30
},
{
"epoch": 0.32,
"learning_rate": 0.0002,
"loss": 2.2569,
"step": 32
},
{
"epoch": 0.34,
"learning_rate": 0.0002,
"loss": 2.1687,
"step": 34
},
{
"epoch": 0.36,
"learning_rate": 0.0002,
"loss": 2.2982,
"step": 36
},
{
"epoch": 0.38,
"learning_rate": 0.0002,
"loss": 2.3297,
"step": 38
},
{
"epoch": 0.4,
"learning_rate": 0.0002,
"loss": 2.2531,
"step": 40
},
{
"epoch": 0.42,
"learning_rate": 0.0002,
"loss": 2.1438,
"step": 42
},
{
"epoch": 0.44,
"learning_rate": 0.0002,
"loss": 1.8699,
"step": 44
},
{
"epoch": 0.46,
"learning_rate": 0.0002,
"loss": 2.175,
"step": 46
},
{
"epoch": 0.48,
"learning_rate": 0.0002,
"loss": 2.0656,
"step": 48
},
{
"epoch": 0.5,
"learning_rate": 0.0002,
"loss": 2.1565,
"step": 50
},
{
"epoch": 0.52,
"learning_rate": 0.0002,
"loss": 2.2622,
"step": 52
},
{
"epoch": 0.54,
"learning_rate": 0.0002,
"loss": 2.1934,
"step": 54
},
{
"epoch": 0.56,
"learning_rate": 0.0002,
"loss": 2.3718,
"step": 56
},
{
"epoch": 0.58,
"learning_rate": 0.0002,
"loss": 2.2122,
"step": 58
},
{
"epoch": 0.6,
"learning_rate": 0.0002,
"loss": 2.25,
"step": 60
},
{
"epoch": 0.62,
"learning_rate": 0.0002,
"loss": 2.2118,
"step": 62
},
{
"epoch": 0.64,
"learning_rate": 0.0002,
"loss": 2.2553,
"step": 64
},
{
"epoch": 0.66,
"learning_rate": 0.0002,
"loss": 2.2255,
"step": 66
},
{
"epoch": 0.68,
"learning_rate": 0.0002,
"loss": 2.1555,
"step": 68
},
{
"epoch": 0.7,
"learning_rate": 0.0002,
"loss": 1.9828,
"step": 70
},
{
"epoch": 0.72,
"learning_rate": 0.0002,
"loss": 2.2447,
"step": 72
},
{
"epoch": 0.74,
"learning_rate": 0.0002,
"loss": 2.0942,
"step": 74
},
{
"epoch": 0.76,
"learning_rate": 0.0002,
"loss": 2.1352,
"step": 76
},
{
"epoch": 0.78,
"learning_rate": 0.0002,
"loss": 2.1952,
"step": 78
},
{
"epoch": 0.8,
"learning_rate": 0.0002,
"loss": 2.3201,
"step": 80
},
{
"epoch": 0.82,
"learning_rate": 0.0002,
"loss": 2.2092,
"step": 82
},
{
"epoch": 0.84,
"learning_rate": 0.0002,
"loss": 2.2096,
"step": 84
},
{
"epoch": 0.86,
"learning_rate": 0.0002,
"loss": 2.3016,
"step": 86
},
{
"epoch": 0.88,
"learning_rate": 0.0002,
"loss": 2.2513,
"step": 88
},
{
"epoch": 0.89,
"learning_rate": 0.0002,
"loss": 2.0121,
"step": 90
},
{
"epoch": 0.91,
"learning_rate": 0.0002,
"loss": 2.068,
"step": 92
},
{
"epoch": 0.93,
"learning_rate": 0.0002,
"loss": 2.0854,
"step": 94
},
{
"epoch": 0.95,
"learning_rate": 0.0002,
"loss": 1.94,
"step": 96
},
{
"epoch": 0.97,
"learning_rate": 0.0002,
"loss": 2.1028,
"step": 98
},
{
"epoch": 0.99,
"learning_rate": 0.0002,
"loss": 1.9648,
"step": 100
}
],
"logging_steps": 2,
"max_steps": 100,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 500,
"total_flos": 8582958605230080.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}