|
{ |
|
"best_metric": 0.6872902512550354, |
|
"best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l57-l/checkpoint-8000", |
|
"epoch": 0.9005966452774964, |
|
"eval_steps": 500, |
|
"global_step": 8000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.056287290329843524, |
|
"grad_norm": 150.1642303466797, |
|
"learning_rate": 4.929640887087696e-07, |
|
"loss": 0.3486, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.056287290329843524, |
|
"eval_loss": 1.1265727281570435, |
|
"eval_runtime": 123.7806, |
|
"eval_samples_per_second": 15.948, |
|
"eval_steps_per_second": 1.995, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.11257458065968705, |
|
"grad_norm": 366.6913146972656, |
|
"learning_rate": 4.859281774175391e-07, |
|
"loss": 0.2733, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.11257458065968705, |
|
"eval_loss": 0.9742079377174377, |
|
"eval_runtime": 125.8047, |
|
"eval_samples_per_second": 15.691, |
|
"eval_steps_per_second": 1.963, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.16886187098953057, |
|
"grad_norm": 555.5016479492188, |
|
"learning_rate": 4.788922661263087e-07, |
|
"loss": 0.1851, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.16886187098953057, |
|
"eval_loss": 0.9162159562110901, |
|
"eval_runtime": 126.7606, |
|
"eval_samples_per_second": 15.573, |
|
"eval_steps_per_second": 1.949, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.2251491613193741, |
|
"grad_norm": 0.423260897397995, |
|
"learning_rate": 4.7185635483507824e-07, |
|
"loss": 0.1973, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.2251491613193741, |
|
"eval_loss": 0.8716973662376404, |
|
"eval_runtime": 127.6955, |
|
"eval_samples_per_second": 15.459, |
|
"eval_steps_per_second": 1.934, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.2814364516492176, |
|
"grad_norm": 6.840139389038086, |
|
"learning_rate": 4.6482044354384774e-07, |
|
"loss": 0.1881, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.2814364516492176, |
|
"eval_loss": 0.830047070980072, |
|
"eval_runtime": 128.0771, |
|
"eval_samples_per_second": 15.413, |
|
"eval_steps_per_second": 1.929, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.33772374197906113, |
|
"grad_norm": 729.8377685546875, |
|
"learning_rate": 4.577845322526173e-07, |
|
"loss": 0.1695, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.33772374197906113, |
|
"eval_loss": 0.8265627026557922, |
|
"eval_runtime": 128.052, |
|
"eval_samples_per_second": 15.416, |
|
"eval_steps_per_second": 1.929, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.39401103230890466, |
|
"grad_norm": 7.96999938756926e-06, |
|
"learning_rate": 4.507486209613869e-07, |
|
"loss": 0.155, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.39401103230890466, |
|
"eval_loss": 0.7721038460731506, |
|
"eval_runtime": 128.1871, |
|
"eval_samples_per_second": 15.399, |
|
"eval_steps_per_second": 1.927, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.4502983226387482, |
|
"grad_norm": 0.006555848754942417, |
|
"learning_rate": 4.4371270967015645e-07, |
|
"loss": 0.1216, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.4502983226387482, |
|
"eval_loss": 0.7490401864051819, |
|
"eval_runtime": 127.5516, |
|
"eval_samples_per_second": 15.476, |
|
"eval_steps_per_second": 1.936, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.5065856129685917, |
|
"grad_norm": 0.0004515685432124883, |
|
"learning_rate": 4.36676798378926e-07, |
|
"loss": 0.1531, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.5065856129685917, |
|
"eval_loss": 0.7533182501792908, |
|
"eval_runtime": 127.4797, |
|
"eval_samples_per_second": 15.485, |
|
"eval_steps_per_second": 1.938, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.5628729032984352, |
|
"grad_norm": 4.4633176003117114e-05, |
|
"learning_rate": 4.2964088708769556e-07, |
|
"loss": 0.1154, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.5628729032984352, |
|
"eval_loss": 0.747530460357666, |
|
"eval_runtime": 127.8675, |
|
"eval_samples_per_second": 15.438, |
|
"eval_steps_per_second": 1.932, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.6191601936282788, |
|
"grad_norm": 4.158839702606201, |
|
"learning_rate": 4.2260497579646517e-07, |
|
"loss": 0.1407, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.6191601936282788, |
|
"eval_loss": 0.724854588508606, |
|
"eval_runtime": 125.1629, |
|
"eval_samples_per_second": 15.771, |
|
"eval_steps_per_second": 1.973, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.6754474839581223, |
|
"grad_norm": 0.748630166053772, |
|
"learning_rate": 4.155690645052347e-07, |
|
"loss": 0.1447, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.6754474839581223, |
|
"eval_loss": 0.7091771960258484, |
|
"eval_runtime": 125.2549, |
|
"eval_samples_per_second": 15.76, |
|
"eval_steps_per_second": 1.972, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.7317347742879657, |
|
"grad_norm": 2.8120924980612472e-05, |
|
"learning_rate": 4.0853315321400427e-07, |
|
"loss": 0.1146, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.7317347742879657, |
|
"eval_loss": 0.7105826735496521, |
|
"eval_runtime": 127.7786, |
|
"eval_samples_per_second": 15.449, |
|
"eval_steps_per_second": 1.933, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.7880220646178093, |
|
"grad_norm": 1.4741635823156685e-07, |
|
"learning_rate": 4.014972419227738e-07, |
|
"loss": 0.171, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.7880220646178093, |
|
"eval_loss": 0.6978534460067749, |
|
"eval_runtime": 125.2567, |
|
"eval_samples_per_second": 15.76, |
|
"eval_steps_per_second": 1.972, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.8443093549476528, |
|
"grad_norm": 0.002975167240947485, |
|
"learning_rate": 3.944613306315434e-07, |
|
"loss": 0.1584, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 0.8443093549476528, |
|
"eval_loss": 0.6923494935035706, |
|
"eval_runtime": 125.3384, |
|
"eval_samples_per_second": 15.749, |
|
"eval_steps_per_second": 1.971, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 0.9005966452774964, |
|
"grad_norm": 9.225498797604814e-05, |
|
"learning_rate": 3.8742541934031293e-07, |
|
"loss": 0.1384, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 0.9005966452774964, |
|
"eval_loss": 0.6872902512550354, |
|
"eval_runtime": 128.114, |
|
"eval_samples_per_second": 15.408, |
|
"eval_steps_per_second": 1.928, |
|
"step": 8000 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 35532, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 4, |
|
"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": 2878786326782520.0, |
|
"train_batch_size": 2, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|