| { |
| "best_global_step": 100, |
| "best_metric": 0.09543673694133759, |
| "best_model_checkpoint": "/content/models/gemma_qlora_lmh/checkpoint-100", |
| "epoch": 1.7008547008547008, |
| "eval_steps": 20, |
| "global_step": 100, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "entropy": 2.4621351063251495, |
| "epoch": 0.3418803418803419, |
| "grad_norm": 6.045567035675049, |
| "learning_rate": 8.389830508474577e-06, |
| "loss": 0.3829, |
| "mean_token_accuracy": 0.8746399849653244, |
| "num_tokens": 113164.0, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.3418803418803419, |
| "eval_entropy": 2.310356734145401, |
| "eval_loss": 0.14163336157798767, |
| "eval_mean_token_accuracy": 0.9525125485200149, |
| "eval_num_tokens": 113164.0, |
| "eval_runtime": 46.7154, |
| "eval_samples_per_second": 39.816, |
| "eval_steps_per_second": 2.505, |
| "step": 20 |
| }, |
| { |
| "entropy": 2.3071940779685973, |
| "epoch": 0.6837606837606838, |
| "grad_norm": 2.0642659664154053, |
| "learning_rate": 6.694915254237288e-06, |
| "loss": 0.1358, |
| "mean_token_accuracy": 0.9574040338397026, |
| "num_tokens": 225335.0, |
| "step": 40 |
| }, |
| { |
| "epoch": 0.6837606837606838, |
| "eval_entropy": 2.279157841307485, |
| "eval_loss": 0.11512647569179535, |
| "eval_mean_token_accuracy": 0.9622599701596122, |
| "eval_num_tokens": 225335.0, |
| "eval_runtime": 45.5615, |
| "eval_samples_per_second": 40.824, |
| "eval_steps_per_second": 2.568, |
| "step": 40 |
| }, |
| { |
| "entropy": 2.3005140133393116, |
| "epoch": 1.017094017094017, |
| "grad_norm": 2.1764779090881348, |
| "learning_rate": 5e-06, |
| "loss": 0.113, |
| "mean_token_accuracy": 0.9657797859265254, |
| "num_tokens": 330390.0, |
| "step": 60 |
| }, |
| { |
| "epoch": 1.017094017094017, |
| "eval_entropy": 2.293168124989567, |
| "eval_loss": 0.10884715616703033, |
| "eval_mean_token_accuracy": 0.9645043224351019, |
| "eval_num_tokens": 330390.0, |
| "eval_runtime": 46.0807, |
| "eval_samples_per_second": 40.364, |
| "eval_steps_per_second": 2.539, |
| "step": 60 |
| }, |
| { |
| "entropy": 2.27779643535614, |
| "epoch": 1.358974358974359, |
| "grad_norm": 2.2281429767608643, |
| "learning_rate": 3.305084745762712e-06, |
| "loss": 0.0849, |
| "mean_token_accuracy": 0.9723166093230248, |
| "num_tokens": 440357.0, |
| "step": 80 |
| }, |
| { |
| "epoch": 1.358974358974359, |
| "eval_entropy": 2.2607989514994826, |
| "eval_loss": 0.10449662059545517, |
| "eval_mean_token_accuracy": 0.9651819151690882, |
| "eval_num_tokens": 440357.0, |
| "eval_runtime": 46.3813, |
| "eval_samples_per_second": 40.102, |
| "eval_steps_per_second": 2.523, |
| "step": 80 |
| }, |
| { |
| "entropy": 2.2726868212223055, |
| "epoch": 1.7008547008547008, |
| "grad_norm": 1.7332485914230347, |
| "learning_rate": 1.6101694915254237e-06, |
| "loss": 0.0719, |
| "mean_token_accuracy": 0.9728455126285553, |
| "num_tokens": 552807.0, |
| "step": 100 |
| }, |
| { |
| "epoch": 1.7008547008547008, |
| "eval_entropy": 2.2467452420128717, |
| "eval_loss": 0.09543673694133759, |
| "eval_mean_token_accuracy": 0.9682549915762029, |
| "eval_num_tokens": 552807.0, |
| "eval_runtime": 46.2095, |
| "eval_samples_per_second": 40.251, |
| "eval_steps_per_second": 2.532, |
| "step": 100 |
| } |
| ], |
| "logging_steps": 20, |
| "max_steps": 118, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 2, |
| "save_steps": 20, |
| "stateful_callbacks": { |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": false |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 1.614440163723264e+16, |
| "train_batch_size": 16, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
