kevinoli's picture
Training in progress, step 5000, checkpoint
e451742 verified
raw
history blame
4.73 kB
{
"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
}