|
{ |
|
"best_metric": 1.0519349575042725, |
|
"best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l58-l/checkpoint-5000", |
|
"epoch": 0.9211495946941783, |
|
"eval_steps": 500, |
|
"global_step": 5000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.09211495946941783, |
|
"grad_norm": 20.04043960571289, |
|
"learning_rate": 4.8848563006632274e-08, |
|
"loss": 0.3758, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.09211495946941783, |
|
"eval_loss": 1.4184927940368652, |
|
"eval_runtime": 74.3562, |
|
"eval_samples_per_second": 16.233, |
|
"eval_steps_per_second": 2.031, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.18422991893883567, |
|
"grad_norm": 74.25025939941406, |
|
"learning_rate": 4.7697126013264556e-08, |
|
"loss": 0.4103, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.18422991893883567, |
|
"eval_loss": 1.3500770330429077, |
|
"eval_runtime": 75.7013, |
|
"eval_samples_per_second": 15.944, |
|
"eval_steps_per_second": 1.995, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.2763448784082535, |
|
"grad_norm": 0.5102410912513733, |
|
"learning_rate": 4.6545689019896826e-08, |
|
"loss": 0.433, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.2763448784082535, |
|
"eval_loss": 1.2885024547576904, |
|
"eval_runtime": 75.197, |
|
"eval_samples_per_second": 16.051, |
|
"eval_steps_per_second": 2.008, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.36845983787767134, |
|
"grad_norm": 0.1667678952217102, |
|
"learning_rate": 4.539425202652911e-08, |
|
"loss": 0.3424, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.36845983787767134, |
|
"eval_loss": 1.239119052886963, |
|
"eval_runtime": 75.387, |
|
"eval_samples_per_second": 16.011, |
|
"eval_steps_per_second": 2.003, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.46057479734708917, |
|
"grad_norm": 343.8175048828125, |
|
"learning_rate": 4.4242815033161385e-08, |
|
"loss": 0.3645, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.46057479734708917, |
|
"eval_loss": 1.1902339458465576, |
|
"eval_runtime": 75.9899, |
|
"eval_samples_per_second": 15.884, |
|
"eval_steps_per_second": 1.987, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.552689756816507, |
|
"grad_norm": 305.5626525878906, |
|
"learning_rate": 4.309137803979366e-08, |
|
"loss": 0.3172, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.552689756816507, |
|
"eval_loss": 1.1506118774414062, |
|
"eval_runtime": 74.8655, |
|
"eval_samples_per_second": 16.122, |
|
"eval_steps_per_second": 2.017, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.6448047162859248, |
|
"grad_norm": 65.59048461914062, |
|
"learning_rate": 4.193994104642594e-08, |
|
"loss": 0.2751, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.6448047162859248, |
|
"eval_loss": 1.1169357299804688, |
|
"eval_runtime": 76.2936, |
|
"eval_samples_per_second": 15.82, |
|
"eval_steps_per_second": 1.979, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.7369196757553427, |
|
"grad_norm": 18.106201171875, |
|
"learning_rate": 4.078850405305821e-08, |
|
"loss": 0.2919, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.7369196757553427, |
|
"eval_loss": 1.0920671224594116, |
|
"eval_runtime": 74.9101, |
|
"eval_samples_per_second": 16.113, |
|
"eval_steps_per_second": 2.016, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.8290346352247605, |
|
"grad_norm": 506.40570068359375, |
|
"learning_rate": 3.9637067059690496e-08, |
|
"loss": 0.2583, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.8290346352247605, |
|
"eval_loss": 1.0721209049224854, |
|
"eval_runtime": 75.8053, |
|
"eval_samples_per_second": 15.922, |
|
"eval_steps_per_second": 1.992, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.9211495946941783, |
|
"grad_norm": 1.1508910655975342, |
|
"learning_rate": 3.848563006632277e-08, |
|
"loss": 0.2679, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.9211495946941783, |
|
"eval_loss": 1.0519349575042725, |
|
"eval_runtime": 75.226, |
|
"eval_samples_per_second": 16.045, |
|
"eval_steps_per_second": 2.007, |
|
"step": 5000 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 21712, |
|
"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": 1799511273002520.0, |
|
"train_batch_size": 2, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|