305 lines
10 KiB
JSON
305 lines
10 KiB
JSON
{
|
|
"best_global_step": null,
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 2.0833333333333335,
|
|
"eval_steps": 500,
|
|
"global_step": 100,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 0.20833333333333334,
|
|
"frac_reward_zero_std": 0.275,
|
|
"grad_norm": 0.252999871969223,
|
|
"kl": 4.623506138159428e-05,
|
|
"learning_rate": 3e-05,
|
|
"loss": 7.562339305877686e-08,
|
|
"num_tokens": 30976.0,
|
|
"reward": -1.63874990940094,
|
|
"reward_std": 0.6745969772338867,
|
|
"rewards/reward_fn/mean": -1.63874990940094,
|
|
"rewards/reward_fn/std": 0.8239436686038971,
|
|
"step": 10
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 0.4166666666666667,
|
|
"frac_reward_zero_std": 0.175,
|
|
"grad_norm": 0.2783929705619812,
|
|
"kl": 0.001219680597569095,
|
|
"learning_rate": 4.8449612403100775e-05,
|
|
"loss": 1.2345612049102782e-06,
|
|
"num_tokens": 62268.0,
|
|
"reward": -1.4524999380111694,
|
|
"reward_std": 0.7632413059473038,
|
|
"rewards/reward_fn/mean": -1.4524999260902405,
|
|
"rewards/reward_fn/std": 0.8776885092258453,
|
|
"step": 20
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 0.625,
|
|
"frac_reward_zero_std": 0.075,
|
|
"grad_norm": 0.310812383890152,
|
|
"kl": 0.012675344076706096,
|
|
"learning_rate": 4.4573643410852715e-05,
|
|
"loss": 1.2670829892158509e-05,
|
|
"num_tokens": 93408.0,
|
|
"reward": -0.9787499606609344,
|
|
"reward_std": 0.8382474899291992,
|
|
"rewards/reward_fn/mean": -0.9787499666213989,
|
|
"rewards/reward_fn/std": 0.874516224861145,
|
|
"step": 30
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 0.8333333333333334,
|
|
"frac_reward_zero_std": 0.075,
|
|
"grad_norm": 0.3383573889732361,
|
|
"kl": 0.03188293152488768,
|
|
"learning_rate": 4.0697674418604655e-05,
|
|
"loss": 3.1878240406513214e-05,
|
|
"num_tokens": 124444.0,
|
|
"reward": -0.7437499940395356,
|
|
"reward_std": 0.7179304540157319,
|
|
"rewards/reward_fn/mean": -0.7437499821186065,
|
|
"rewards/reward_fn/std": 0.7846309900283813,
|
|
"step": 40
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 1.0416666666666667,
|
|
"frac_reward_zero_std": 0.225,
|
|
"grad_norm": 0.35372963547706604,
|
|
"kl": 0.04432902028784156,
|
|
"learning_rate": 3.682170542635659e-05,
|
|
"loss": 4.432052373886109e-05,
|
|
"num_tokens": 155440.0,
|
|
"reward": -0.6174999922513962,
|
|
"reward_std": 0.5982875227928162,
|
|
"rewards/reward_fn/mean": -0.6174999952316285,
|
|
"rewards/reward_fn/std": 0.698276162147522,
|
|
"step": 50
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 1.25,
|
|
"frac_reward_zero_std": 0.475,
|
|
"grad_norm": 0.19890302419662476,
|
|
"kl": 0.06396519551053644,
|
|
"learning_rate": 3.294573643410852e-05,
|
|
"loss": 6.395205855369568e-05,
|
|
"num_tokens": 186420.0,
|
|
"reward": -0.4387499988079071,
|
|
"reward_std": 0.3215150013566017,
|
|
"rewards/reward_fn/mean": -0.4387499988079071,
|
|
"rewards/reward_fn/std": 0.48022162914276123,
|
|
"step": 60
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 1.4583333333333333,
|
|
"frac_reward_zero_std": 0.5,
|
|
"grad_norm": 0.3041526675224304,
|
|
"kl": 0.08288784278556705,
|
|
"learning_rate": 2.9069767441860467e-05,
|
|
"loss": 8.288174867630005e-05,
|
|
"num_tokens": 217644.0,
|
|
"reward": -0.3737500041723251,
|
|
"reward_std": 0.2771410092711449,
|
|
"rewards/reward_fn/mean": -0.3737500011920929,
|
|
"rewards/reward_fn/std": 0.4215263336896896,
|
|
"step": 70
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 1.6666666666666665,
|
|
"frac_reward_zero_std": 0.725,
|
|
"grad_norm": 0.3231094479560852,
|
|
"kl": 0.10231811739504337,
|
|
"learning_rate": 2.5193798449612404e-05,
|
|
"loss": 0.00010231174528598785,
|
|
"num_tokens": 248780.0,
|
|
"reward": -0.34500000476837156,
|
|
"reward_std": 0.17121320217847824,
|
|
"rewards/reward_fn/mean": -0.34500000476837156,
|
|
"rewards/reward_fn/std": 0.26369245946407316,
|
|
"step": 80
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 1.875,
|
|
"frac_reward_zero_std": 0.675,
|
|
"grad_norm": 0.3435279428958893,
|
|
"kl": 0.123704750277102,
|
|
"learning_rate": 2.131782945736434e-05,
|
|
"loss": 0.0001236904412508011,
|
|
"num_tokens": 279836.0,
|
|
"reward": -0.3937500029802322,
|
|
"reward_std": 0.18749999403953552,
|
|
"rewards/reward_fn/mean": -0.3937500029802322,
|
|
"rewards/reward_fn/std": 0.29925094842910765,
|
|
"step": 90
|
|
},
|
|
{
|
|
"clip_ratio/high_max": 0.0,
|
|
"clip_ratio/high_mean": 0.0,
|
|
"clip_ratio/low_mean": 0.0,
|
|
"clip_ratio/low_min": 0.0,
|
|
"clip_ratio/region_mean": 0.0,
|
|
"completion_length": 100.0,
|
|
"completions/clipped_ratio": 1.0,
|
|
"completions/max_length": 100.0,
|
|
"completions/max_terminated_length": 0.0,
|
|
"completions/mean_length": 100.0,
|
|
"completions/mean_terminated_length": 0.0,
|
|
"completions/min_length": 100.0,
|
|
"completions/min_terminated_length": 0.0,
|
|
"epoch": 2.0833333333333335,
|
|
"frac_reward_zero_std": 0.725,
|
|
"grad_norm": 0.0014862708048895001,
|
|
"kl": 0.1360227341763675,
|
|
"learning_rate": 1.744186046511628e-05,
|
|
"loss": 0.0001360199647024274,
|
|
"num_tokens": 310908.0,
|
|
"reward": -0.33250000327825546,
|
|
"reward_std": 0.20633567869663239,
|
|
"rewards/reward_fn/mean": -0.3325000062584877,
|
|
"rewards/reward_fn/std": 0.3232262939214706,
|
|
"step": 100
|
|
}
|
|
],
|
|
"logging_steps": 10,
|
|
"max_steps": 144,
|
|
"num_input_tokens_seen": 310908,
|
|
"num_train_epochs": 3,
|
|
"save_steps": 50,
|
|
"stateful_callbacks": {
|
|
"TrainerControl": {
|
|
"args": {
|
|
"should_epoch_stop": false,
|
|
"should_evaluate": false,
|
|
"should_log": false,
|
|
"should_save": true,
|
|
"should_training_stop": false
|
|
},
|
|
"attributes": {}
|
|
}
|
|
},
|
|
"total_flos": 0.0,
|
|
"train_batch_size": 4,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|