| { |
| "best_global_step": 200, |
| "best_metric": 0.24931415915489197, |
| "best_model_checkpoint": "/workspace/andy-D1-andyft/checkpoint-200", |
| "epoch": 0.3679006668199586, |
| "eval_steps": 100, |
| "global_step": 200, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "entropy": 1.358754662796855, |
| "epoch": 0.03679006668199586, |
| "grad_norm": 0.322265625, |
| "learning_rate": 6.785714285714286e-05, |
| "loss": 1.9387958526611329, |
| "mean_token_accuracy": 0.6328379280865193, |
| "num_tokens": 552605.0, |
| "step": 20 |
| }, |
| { |
| "entropy": 1.2090021014213561, |
| "epoch": 0.07358013336399172, |
| "grad_norm": 0.3359375, |
| "learning_rate": 9.988791089826816e-05, |
| "loss": 1.0881237983703613, |
| "mean_token_accuracy": 0.7621329009532929, |
| "num_tokens": 1118812.0, |
| "step": 40 |
| }, |
| { |
| "entropy": 0.5365577314049006, |
| "epoch": 0.11037020004598758, |
| "grad_norm": 0.265625, |
| "learning_rate": 9.91120793909987e-05, |
| "loss": 0.5044384002685547, |
| "mean_token_accuracy": 0.8858215417712927, |
| "num_tokens": 1693574.0, |
| "step": 60 |
| }, |
| { |
| "entropy": 0.4165896608028561, |
| "epoch": 0.14716026672798344, |
| "grad_norm": 0.2451171875, |
| "learning_rate": 9.760895058620235e-05, |
| "loss": 0.38671829700469973, |
| "mean_token_accuracy": 0.9098233781754971, |
| "num_tokens": 2267338.0, |
| "step": 80 |
| }, |
| { |
| "entropy": 0.3438017311505973, |
| "epoch": 0.1839503334099793, |
| "grad_norm": 0.2197265625, |
| "learning_rate": 9.540078421165121e-05, |
| "loss": 0.3093204736709595, |
| "mean_token_accuracy": 0.9240799810737371, |
| "num_tokens": 2850973.0, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.1839503334099793, |
| "eval_entropy": 0.33466541236795194, |
| "eval_loss": 0.3281489610671997, |
| "eval_mean_token_accuracy": 0.9268425211115175, |
| "eval_num_tokens": 2850973.0, |
| "eval_runtime": 156.4612, |
| "eval_samples_per_second": 2.921, |
| "eval_steps_per_second": 1.464, |
| "step": 100 |
| }, |
| { |
| "entropy": 0.3134843219071627, |
| "epoch": 0.22074040009197515, |
| "grad_norm": 0.189453125, |
| "learning_rate": 9.252028084644137e-05, |
| "loss": 0.28039026260375977, |
| "mean_token_accuracy": 0.92984459400177, |
| "num_tokens": 3429327.0, |
| "step": 120 |
| }, |
| { |
| "entropy": 0.3144511237274855, |
| "epoch": 0.257530466773971, |
| "grad_norm": 0.22265625, |
| "learning_rate": 8.901009766043847e-05, |
| "loss": 0.27316670417785643, |
| "mean_token_accuracy": 0.9302996207028628, |
| "num_tokens": 3995502.0, |
| "step": 140 |
| }, |
| { |
| "entropy": 0.28906950131058695, |
| "epoch": 0.29432053345596687, |
| "grad_norm": 0.267578125, |
| "learning_rate": 8.492221670727694e-05, |
| "loss": 0.2562049150466919, |
| "mean_token_accuracy": 0.9345210112631321, |
| "num_tokens": 4561185.0, |
| "step": 160 |
| }, |
| { |
| "entropy": 0.309068296616897, |
| "epoch": 0.33111060013796273, |
| "grad_norm": 0.255859375, |
| "learning_rate": 8.031717512581784e-05, |
| "loss": 0.2678884267807007, |
| "mean_token_accuracy": 0.9299660053104162, |
| "num_tokens": 5133715.0, |
| "step": 180 |
| }, |
| { |
| "entropy": 0.27301658713258803, |
| "epoch": 0.3679006668199586, |
| "grad_norm": 0.35546875, |
| "learning_rate": 7.526316864995647e-05, |
| "loss": 0.23800337314605713, |
| "mean_token_accuracy": 0.938116031512618, |
| "num_tokens": 5698771.0, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.3679006668199586, |
| "eval_entropy": 0.261124916667501, |
| "eval_loss": 0.24931415915489197, |
| "eval_mean_token_accuracy": 0.9429609785954505, |
| "eval_num_tokens": 5698771.0, |
| "eval_runtime": 156.272, |
| "eval_samples_per_second": 2.924, |
| "eval_steps_per_second": 1.465, |
| "step": 200 |
| } |
| ], |
| "logging_steps": 20, |
| "max_steps": 544, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 1, |
| "save_steps": 100, |
| "stateful_callbacks": { |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": false |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 2.7278687962629734e+17, |
| "train_batch_size": 2, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
