|
{ |
|
"best_metric": 0.6204637289047241, |
|
"best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l57-l/checkpoint-7000", |
|
"epoch": 0.7465870307167235, |
|
"eval_steps": 500, |
|
"global_step": 7000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.05332764505119454, |
|
"grad_norm": 414.2552795410156, |
|
"learning_rate": 4.911120591581342e-07, |
|
"loss": 0.3809, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.05332764505119454, |
|
"eval_loss": 1.1109352111816406, |
|
"eval_runtime": 63.6734, |
|
"eval_samples_per_second": 15.501, |
|
"eval_steps_per_second": 1.947, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.10665529010238908, |
|
"grad_norm": 45.880027770996094, |
|
"learning_rate": 4.822241183162685e-07, |
|
"loss": 0.2649, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.10665529010238908, |
|
"eval_loss": 0.9588962197303772, |
|
"eval_runtime": 62.6825, |
|
"eval_samples_per_second": 15.746, |
|
"eval_steps_per_second": 1.978, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.1599829351535836, |
|
"grad_norm": 432.9915466308594, |
|
"learning_rate": 4.733361774744027e-07, |
|
"loss": 0.2104, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.1599829351535836, |
|
"eval_loss": 0.90110844373703, |
|
"eval_runtime": 62.6757, |
|
"eval_samples_per_second": 15.748, |
|
"eval_steps_per_second": 1.978, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.21331058020477817, |
|
"grad_norm": 0.014208819717168808, |
|
"learning_rate": 4.6444823663253695e-07, |
|
"loss": 0.1849, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.21331058020477817, |
|
"eval_loss": 0.8570474982261658, |
|
"eval_runtime": 63.7296, |
|
"eval_samples_per_second": 15.487, |
|
"eval_steps_per_second": 1.946, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.2666382252559727, |
|
"grad_norm": 174.73324584960938, |
|
"learning_rate": 4.5556029579067116e-07, |
|
"loss": 0.2056, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.2666382252559727, |
|
"eval_loss": 0.7895939350128174, |
|
"eval_runtime": 63.8387, |
|
"eval_samples_per_second": 15.461, |
|
"eval_steps_per_second": 1.942, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.3199658703071672, |
|
"grad_norm": 9.852073823424234e-09, |
|
"learning_rate": 4.4667235494880547e-07, |
|
"loss": 0.159, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.3199658703071672, |
|
"eval_loss": 0.7647623419761658, |
|
"eval_runtime": 63.9155, |
|
"eval_samples_per_second": 15.442, |
|
"eval_steps_per_second": 1.94, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.37329351535836175, |
|
"grad_norm": 0.0025215104687958956, |
|
"learning_rate": 4.377844141069397e-07, |
|
"loss": 0.1696, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.37329351535836175, |
|
"eval_loss": 0.7638036012649536, |
|
"eval_runtime": 63.8003, |
|
"eval_samples_per_second": 15.47, |
|
"eval_steps_per_second": 1.944, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.42662116040955633, |
|
"grad_norm": 2.4611830711364746, |
|
"learning_rate": 4.2889647326507393e-07, |
|
"loss": 0.1125, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.42662116040955633, |
|
"eval_loss": 0.7486374378204346, |
|
"eval_runtime": 63.9847, |
|
"eval_samples_per_second": 15.426, |
|
"eval_steps_per_second": 1.938, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.47994880546075086, |
|
"grad_norm": 5.230295658111572, |
|
"learning_rate": 4.2000853242320814e-07, |
|
"loss": 0.1322, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.47994880546075086, |
|
"eval_loss": 0.7251659035682678, |
|
"eval_runtime": 64.4782, |
|
"eval_samples_per_second": 15.308, |
|
"eval_steps_per_second": 1.923, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.5332764505119454, |
|
"grad_norm": 3.821002974291332e-05, |
|
"learning_rate": 4.1112059158134245e-07, |
|
"loss": 0.1633, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.5332764505119454, |
|
"eval_loss": 0.7035414576530457, |
|
"eval_runtime": 62.6186, |
|
"eval_samples_per_second": 15.762, |
|
"eval_steps_per_second": 1.98, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.58660409556314, |
|
"grad_norm": 5.916939244343666e-06, |
|
"learning_rate": 4.0223265073947665e-07, |
|
"loss": 0.125, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.58660409556314, |
|
"eval_loss": 0.6792380809783936, |
|
"eval_runtime": 65.0551, |
|
"eval_samples_per_second": 15.172, |
|
"eval_steps_per_second": 1.906, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.6399317406143344, |
|
"grad_norm": 92.8386459350586, |
|
"learning_rate": 3.933447098976109e-07, |
|
"loss": 0.1297, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.6399317406143344, |
|
"eval_loss": 0.6549482345581055, |
|
"eval_runtime": 64.0827, |
|
"eval_samples_per_second": 15.402, |
|
"eval_steps_per_second": 1.935, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.693259385665529, |
|
"grad_norm": 0.00014717792510055006, |
|
"learning_rate": 3.8445676905574517e-07, |
|
"loss": 0.1114, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.693259385665529, |
|
"eval_loss": 0.6223254799842834, |
|
"eval_runtime": 64.89, |
|
"eval_samples_per_second": 15.21, |
|
"eval_steps_per_second": 1.911, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.7465870307167235, |
|
"grad_norm": 4.789559397977428e-07, |
|
"learning_rate": 3.755688282138794e-07, |
|
"loss": 0.144, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.7465870307167235, |
|
"eval_loss": 0.6204637289047241, |
|
"eval_runtime": 62.7674, |
|
"eval_samples_per_second": 15.725, |
|
"eval_steps_per_second": 1.976, |
|
"step": 7000 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 28128, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 3, |
|
"save_steps": 500, |
|
"stateful_callbacks": { |
|
"TrainerControl": { |
|
"args": { |
|
"should_epoch_stop": false, |
|
"should_evaluate": false, |
|
"should_log": false, |
|
"should_save": true, |
|
"should_training_stop": false |
|
}, |
|
"attributes": {} |
|
} |
|
}, |
|
"total_flos": 2519027975522520.0, |
|
"train_batch_size": 2, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|