vit-base-gpu / trainer_state.json
jerlawson13's picture
Training in progress, step 100
a07f2f2 verified
raw
history blame
No virus
6.37 kB
{
"best_metric": 0.10813982784748077,
"best_model_checkpoint": "/home/projects/Project - Fundus Photo Glaucoma Detection (AI)//model_results/vit-base-gpu/checkpoint-200",
"epoch": 5.0,
"eval_steps": 100,
"global_step": 285,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.17543859649122806,
"grad_norm": 1.040771245956421,
"learning_rate": 9.649122807017545e-06,
"loss": 0.6273,
"step": 10
},
{
"epoch": 0.3508771929824561,
"grad_norm": 2.329632520675659,
"learning_rate": 9.298245614035088e-06,
"loss": 0.5016,
"step": 20
},
{
"epoch": 0.5263157894736842,
"grad_norm": 1.6268341541290283,
"learning_rate": 8.947368421052632e-06,
"loss": 0.3763,
"step": 30
},
{
"epoch": 0.7017543859649122,
"grad_norm": 2.5890719890594482,
"learning_rate": 8.596491228070176e-06,
"loss": 0.2924,
"step": 40
},
{
"epoch": 0.8771929824561403,
"grad_norm": 0.7261876463890076,
"learning_rate": 8.24561403508772e-06,
"loss": 0.2481,
"step": 50
},
{
"epoch": 1.0526315789473684,
"grad_norm": 0.8194475769996643,
"learning_rate": 7.894736842105265e-06,
"loss": 0.1682,
"step": 60
},
{
"epoch": 1.2280701754385965,
"grad_norm": 0.5261658430099487,
"learning_rate": 7.5438596491228074e-06,
"loss": 0.1543,
"step": 70
},
{
"epoch": 1.4035087719298245,
"grad_norm": 0.5462292432785034,
"learning_rate": 7.192982456140352e-06,
"loss": 0.1492,
"step": 80
},
{
"epoch": 1.5789473684210527,
"grad_norm": 0.6172283887863159,
"learning_rate": 6.842105263157896e-06,
"loss": 0.1214,
"step": 90
},
{
"epoch": 1.7543859649122808,
"grad_norm": 0.48490244150161743,
"learning_rate": 6.491228070175439e-06,
"loss": 0.1209,
"step": 100
},
{
"epoch": 1.7543859649122808,
"eval_loss": 0.15530365705490112,
"eval_runtime": 24.2994,
"eval_samples_per_second": 9.342,
"eval_steps_per_second": 0.329,
"step": 100
},
{
"epoch": 1.9298245614035088,
"grad_norm": 6.183306694030762,
"learning_rate": 6.140350877192983e-06,
"loss": 0.1262,
"step": 110
},
{
"epoch": 2.1052631578947367,
"grad_norm": 0.6162407994270325,
"learning_rate": 5.789473684210527e-06,
"loss": 0.1122,
"step": 120
},
{
"epoch": 2.280701754385965,
"grad_norm": 0.4731535315513611,
"learning_rate": 5.438596491228071e-06,
"loss": 0.1169,
"step": 130
},
{
"epoch": 2.456140350877193,
"grad_norm": 0.3595629334449768,
"learning_rate": 5.087719298245615e-06,
"loss": 0.0814,
"step": 140
},
{
"epoch": 2.6315789473684212,
"grad_norm": 0.3399164080619812,
"learning_rate": 4.736842105263158e-06,
"loss": 0.0914,
"step": 150
},
{
"epoch": 2.807017543859649,
"grad_norm": 2.8990585803985596,
"learning_rate": 4.385964912280702e-06,
"loss": 0.0856,
"step": 160
},
{
"epoch": 2.982456140350877,
"grad_norm": 5.240866184234619,
"learning_rate": 4.035087719298246e-06,
"loss": 0.0874,
"step": 170
},
{
"epoch": 3.1578947368421053,
"grad_norm": 0.3441706597805023,
"learning_rate": 3.6842105263157896e-06,
"loss": 0.0895,
"step": 180
},
{
"epoch": 3.3333333333333335,
"grad_norm": 0.3535279631614685,
"learning_rate": 3.3333333333333333e-06,
"loss": 0.0881,
"step": 190
},
{
"epoch": 3.5087719298245617,
"grad_norm": 0.6653788685798645,
"learning_rate": 2.9824561403508774e-06,
"loss": 0.0911,
"step": 200
},
{
"epoch": 3.5087719298245617,
"eval_loss": 0.10813982784748077,
"eval_runtime": 29.8414,
"eval_samples_per_second": 7.607,
"eval_steps_per_second": 0.268,
"step": 200
},
{
"epoch": 3.6842105263157894,
"grad_norm": 0.271210640668869,
"learning_rate": 2.631578947368421e-06,
"loss": 0.1006,
"step": 210
},
{
"epoch": 3.8596491228070176,
"grad_norm": 0.35550448298454285,
"learning_rate": 2.280701754385965e-06,
"loss": 0.0703,
"step": 220
},
{
"epoch": 4.035087719298246,
"grad_norm": 0.5986644625663757,
"learning_rate": 1.929824561403509e-06,
"loss": 0.1125,
"step": 230
},
{
"epoch": 4.2105263157894735,
"grad_norm": 0.29196077585220337,
"learning_rate": 1.5789473684210526e-06,
"loss": 0.076,
"step": 240
},
{
"epoch": 4.385964912280702,
"grad_norm": 0.2573108673095703,
"learning_rate": 1.2280701754385965e-06,
"loss": 0.0574,
"step": 250
},
{
"epoch": 4.56140350877193,
"grad_norm": 0.2528444230556488,
"learning_rate": 8.771929824561404e-07,
"loss": 0.0565,
"step": 260
},
{
"epoch": 4.7368421052631575,
"grad_norm": 0.25505849719047546,
"learning_rate": 5.263157894736843e-07,
"loss": 0.0498,
"step": 270
},
{
"epoch": 4.912280701754386,
"grad_norm": 0.541071891784668,
"learning_rate": 1.7543859649122808e-07,
"loss": 0.0945,
"step": 280
},
{
"epoch": 5.0,
"step": 285,
"total_flos": 7.020774259082035e+17,
"train_loss": 0.15416914590618067,
"train_runtime": 2897.051,
"train_samples_per_second": 3.148,
"train_steps_per_second": 0.098
}
],
"logging_steps": 10,
"max_steps": 285,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"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": 7.020774259082035e+17,
"train_batch_size": 32,
"trial_name": null,
"trial_params": null
}