| { | |
| "best_metric": 0.0019312179647386074, | |
| "best_model_checkpoint": "./vision_transformer_model_progress/checkpoint-200", | |
| "epoch": 9.0, | |
| "eval_steps": 100, | |
| "global_step": 297, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.30303030303030304, | |
| "grad_norm": 0.12155108898878098, | |
| "learning_rate": 2.0000000000000003e-06, | |
| "loss": 0.0158, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.6060606060606061, | |
| "grad_norm": 1.0541080236434937, | |
| "learning_rate": 4.000000000000001e-06, | |
| "loss": 0.0479, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.9090909090909091, | |
| "grad_norm": 0.03966180980205536, | |
| "learning_rate": 6e-06, | |
| "loss": 0.0047, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 1.2121212121212122, | |
| "grad_norm": 0.06989104300737381, | |
| "learning_rate": 8.000000000000001e-06, | |
| "loss": 0.0045, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 1.5151515151515151, | |
| "grad_norm": 0.0959181860089302, | |
| "learning_rate": 1e-05, | |
| "loss": 0.0027, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.8181818181818183, | |
| "grad_norm": 0.05803044140338898, | |
| "learning_rate": 1.2e-05, | |
| "loss": 0.0281, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 2.121212121212121, | |
| "grad_norm": 0.24542172253131866, | |
| "learning_rate": 1.4000000000000001e-05, | |
| "loss": 0.0049, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 2.4242424242424243, | |
| "grad_norm": 0.0824781134724617, | |
| "learning_rate": 1.6000000000000003e-05, | |
| "loss": 0.0018, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 2.7272727272727275, | |
| "grad_norm": 0.05576428771018982, | |
| "learning_rate": 1.8e-05, | |
| "loss": 0.0015, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 3.0303030303030303, | |
| "grad_norm": 0.02234538644552231, | |
| "learning_rate": 2e-05, | |
| "loss": 0.0018, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 3.0303030303030303, | |
| "eval_loss": 0.0030471435748040676, | |
| "eval_runtime": 12.9149, | |
| "eval_samples_per_second": 10.066, | |
| "eval_steps_per_second": 0.697, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 3.3333333333333335, | |
| "grad_norm": 0.014021256007254124, | |
| "learning_rate": 2.2000000000000003e-05, | |
| "loss": 0.001, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 3.6363636363636362, | |
| "grad_norm": 0.03260282799601555, | |
| "learning_rate": 2.4e-05, | |
| "loss": 0.0011, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 3.9393939393939394, | |
| "grad_norm": 0.020618194714188576, | |
| "learning_rate": 2.6000000000000002e-05, | |
| "loss": 0.0012, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 4.242424242424242, | |
| "grad_norm": 0.021722646430134773, | |
| "learning_rate": 2.8000000000000003e-05, | |
| "loss": 0.0009, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 4.545454545454545, | |
| "grad_norm": 0.015248863026499748, | |
| "learning_rate": 3e-05, | |
| "loss": 0.0007, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 4.848484848484849, | |
| "grad_norm": 0.00668317498639226, | |
| "learning_rate": 3.2000000000000005e-05, | |
| "loss": 0.0007, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 5.151515151515151, | |
| "grad_norm": 0.009767434559762478, | |
| "learning_rate": 3.4000000000000007e-05, | |
| "loss": 0.0006, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 5.454545454545454, | |
| "grad_norm": 0.0070835500955581665, | |
| "learning_rate": 3.6e-05, | |
| "loss": 0.0005, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 5.757575757575758, | |
| "grad_norm": 0.005514961667358875, | |
| "learning_rate": 3.8e-05, | |
| "loss": 0.0005, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 6.0606060606060606, | |
| "grad_norm": 0.007456360850483179, | |
| "learning_rate": 4e-05, | |
| "loss": 0.0005, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 6.0606060606060606, | |
| "eval_loss": 0.0019312179647386074, | |
| "eval_runtime": 13.4333, | |
| "eval_samples_per_second": 9.677, | |
| "eval_steps_per_second": 0.67, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 6.363636363636363, | |
| "grad_norm": 0.0049119917675852776, | |
| "learning_rate": 4.2e-05, | |
| "loss": 0.0004, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 6.666666666666667, | |
| "grad_norm": 0.0046443757601082325, | |
| "learning_rate": 4.4000000000000006e-05, | |
| "loss": 0.0004, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 6.96969696969697, | |
| "grad_norm": 0.006061806809157133, | |
| "learning_rate": 4.600000000000001e-05, | |
| "loss": 0.0004, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 7.2727272727272725, | |
| "grad_norm": 0.004031210206449032, | |
| "learning_rate": 4.8e-05, | |
| "loss": 0.0004, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 7.575757575757576, | |
| "grad_norm": 0.004312009084969759, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0003, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 7.878787878787879, | |
| "grad_norm": 0.004320390522480011, | |
| "learning_rate": 5.2000000000000004e-05, | |
| "loss": 0.0003, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 8.181818181818182, | |
| "grad_norm": 0.0036691236309707165, | |
| "learning_rate": 5.4000000000000005e-05, | |
| "loss": 0.0003, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 8.484848484848484, | |
| "grad_norm": 0.0040830825455486774, | |
| "learning_rate": 5.6000000000000006e-05, | |
| "loss": 0.0003, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 8.787878787878787, | |
| "grad_norm": 0.003878690768033266, | |
| "learning_rate": 5.8e-05, | |
| "loss": 0.0002, | |
| "step": 290 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 297, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 9, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 3.62740521701376e+17, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |