| { | |
| "best_metric": 0.08870936185121536, | |
| "best_model_checkpoint": "/home1/datawork/mcontini/models/multilabel/huggingface/dinov2-large-2024_01_15-with_data_aug_batch-size32_epochs20_freeze/checkpoint-5480", | |
| "epoch": 20.0, | |
| "eval_steps": 500, | |
| "global_step": 5480, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.4727843684577809, | |
| "eval_f1_macro": 0.5944419296607243, | |
| "eval_f1_micro": 0.7681334483750359, | |
| "eval_loss": 0.13637755811214447, | |
| "eval_roc_auc": 0.8536429092663866, | |
| "eval_runtime": 697.0579, | |
| "eval_samples_per_second": 4.112, | |
| "eval_steps_per_second": 0.129, | |
| "learning_rate": 0.001, | |
| "step": 274 | |
| }, | |
| { | |
| "epoch": 1.82, | |
| "learning_rate": 0.001, | |
| "loss": 0.2417, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.5027913468248429, | |
| "eval_f1_macro": 0.6565191014600725, | |
| "eval_f1_micro": 0.8039716641505052, | |
| "eval_loss": 0.11179632693529129, | |
| "eval_roc_auc": 0.8700912167534057, | |
| "eval_runtime": 694.482, | |
| "eval_samples_per_second": 4.127, | |
| "eval_steps_per_second": 0.13, | |
| "learning_rate": 0.001, | |
| "step": 548 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.5122121423586881, | |
| "eval_f1_macro": 0.6696761965075133, | |
| "eval_f1_micro": 0.8123808973840735, | |
| "eval_loss": 0.10605615377426147, | |
| "eval_roc_auc": 0.8762780072894476, | |
| "eval_runtime": 690.0852, | |
| "eval_samples_per_second": 4.153, | |
| "eval_steps_per_second": 0.13, | |
| "learning_rate": 0.001, | |
| "step": 822 | |
| }, | |
| { | |
| "epoch": 3.65, | |
| "learning_rate": 0.001, | |
| "loss": 0.1314, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.5003489183531054, | |
| "eval_f1_macro": 0.6643012293953361, | |
| "eval_f1_micro": 0.8167795548978313, | |
| "eval_loss": 0.1062413826584816, | |
| "eval_roc_auc": 0.883174651226497, | |
| "eval_runtime": 689.9525, | |
| "eval_samples_per_second": 4.154, | |
| "eval_steps_per_second": 0.13, | |
| "learning_rate": 0.001, | |
| "step": 1096 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.5177948360083741, | |
| "eval_f1_macro": 0.6735692510804457, | |
| "eval_f1_micro": 0.8176450156304272, | |
| "eval_loss": 0.1032244861125946, | |
| "eval_roc_auc": 0.8783293038976828, | |
| "eval_runtime": 692.1367, | |
| "eval_samples_per_second": 4.141, | |
| "eval_steps_per_second": 0.13, | |
| "learning_rate": 0.001, | |
| "step": 1370 | |
| }, | |
| { | |
| "epoch": 5.47, | |
| "learning_rate": 0.001, | |
| "loss": 0.1235, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.5334961618981159, | |
| "eval_f1_macro": 0.6928209901104907, | |
| "eval_f1_micro": 0.8254685030480923, | |
| "eval_loss": 0.10265211015939713, | |
| "eval_roc_auc": 0.890592629363693, | |
| "eval_runtime": 693.5115, | |
| "eval_samples_per_second": 4.133, | |
| "eval_steps_per_second": 0.13, | |
| "learning_rate": 0.001, | |
| "step": 1644 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.5237264480111654, | |
| "eval_f1_macro": 0.6766580498538523, | |
| "eval_f1_micro": 0.8204888862960155, | |
| "eval_loss": 0.10273609310388565, | |
| "eval_roc_auc": 0.8774435954012739, | |
| "eval_runtime": 687.3351, | |
| "eval_samples_per_second": 4.17, | |
| "eval_steps_per_second": 0.131, | |
| "learning_rate": 0.001, | |
| "step": 1918 | |
| }, | |
| { | |
| "epoch": 7.3, | |
| "learning_rate": 0.001, | |
| "loss": 0.1196, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.5181437543614794, | |
| "eval_f1_macro": 0.6757509958691195, | |
| "eval_f1_micro": 0.8175961929081308, | |
| "eval_loss": 0.10268854349851608, | |
| "eval_roc_auc": 0.877528234320222, | |
| "eval_runtime": 684.4409, | |
| "eval_samples_per_second": 4.187, | |
| "eval_steps_per_second": 0.131, | |
| "learning_rate": 0.001, | |
| "step": 2192 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "eval_accuracy": 0.5334961618981159, | |
| "eval_f1_macro": 0.6807132608736285, | |
| "eval_f1_micro": 0.8223761446348907, | |
| "eval_loss": 0.09944070875644684, | |
| "eval_roc_auc": 0.8765387465085724, | |
| "eval_runtime": 676.6666, | |
| "eval_samples_per_second": 4.235, | |
| "eval_steps_per_second": 0.133, | |
| "learning_rate": 0.001, | |
| "step": 2466 | |
| }, | |
| { | |
| "epoch": 9.12, | |
| "learning_rate": 0.001, | |
| "loss": 0.117, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy": 0.5167480809490579, | |
| "eval_f1_macro": 0.6870451830571112, | |
| "eval_f1_micro": 0.8282680748272568, | |
| "eval_loss": 0.1007496640086174, | |
| "eval_roc_auc": 0.8936902868676632, | |
| "eval_runtime": 683.7537, | |
| "eval_samples_per_second": 4.192, | |
| "eval_steps_per_second": 0.132, | |
| "learning_rate": 0.001, | |
| "step": 2740 | |
| }, | |
| { | |
| "epoch": 10.95, | |
| "learning_rate": 0.001, | |
| "loss": 0.1163, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 11.0, | |
| "eval_accuracy": 0.5195394277739009, | |
| "eval_f1_macro": 0.692531553915217, | |
| "eval_f1_micro": 0.8297520661157025, | |
| "eval_loss": 0.09709072858095169, | |
| "eval_roc_auc": 0.8897993342199938, | |
| "eval_runtime": 686.7241, | |
| "eval_samples_per_second": 4.173, | |
| "eval_steps_per_second": 0.131, | |
| "learning_rate": 0.001, | |
| "step": 3014 | |
| }, | |
| { | |
| "epoch": 12.0, | |
| "eval_accuracy": 0.5230286113049546, | |
| "eval_f1_macro": 0.7005826976314891, | |
| "eval_f1_micro": 0.8281771132834963, | |
| "eval_loss": 0.09867899119853973, | |
| "eval_roc_auc": 0.8860984441208896, | |
| "eval_runtime": 684.6008, | |
| "eval_samples_per_second": 4.186, | |
| "eval_steps_per_second": 0.131, | |
| "learning_rate": 0.001, | |
| "step": 3288 | |
| }, | |
| { | |
| "epoch": 12.77, | |
| "learning_rate": 0.001, | |
| "loss": 0.1156, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 13.0, | |
| "eval_accuracy": 0.5341939986043266, | |
| "eval_f1_macro": 0.7064865780322945, | |
| "eval_f1_micro": 0.8274845512784171, | |
| "eval_loss": 0.10170716792345047, | |
| "eval_roc_auc": 0.8902887196802852, | |
| "eval_runtime": 690.7422, | |
| "eval_samples_per_second": 4.149, | |
| "eval_steps_per_second": 0.13, | |
| "learning_rate": 0.001, | |
| "step": 3562 | |
| }, | |
| { | |
| "epoch": 14.0, | |
| "eval_accuracy": 0.5275645498953245, | |
| "eval_f1_macro": 0.696836574892494, | |
| "eval_f1_micro": 0.8243181297272519, | |
| "eval_loss": 0.12239944934844971, | |
| "eval_roc_auc": 0.8851480881066337, | |
| "eval_runtime": 698.6447, | |
| "eval_samples_per_second": 4.102, | |
| "eval_steps_per_second": 0.129, | |
| "learning_rate": 0.001, | |
| "step": 3836 | |
| }, | |
| { | |
| "epoch": 14.6, | |
| "learning_rate": 0.001, | |
| "loss": 0.1137, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 15.0, | |
| "eval_accuracy": 0.5300069783670621, | |
| "eval_f1_macro": 0.6958370181036179, | |
| "eval_f1_micro": 0.8294763177346904, | |
| "eval_loss": 0.09806440025568008, | |
| "eval_roc_auc": 0.8904217072114206, | |
| "eval_runtime": 695.7592, | |
| "eval_samples_per_second": 4.119, | |
| "eval_steps_per_second": 0.129, | |
| "learning_rate": 0.001, | |
| "step": 4110 | |
| }, | |
| { | |
| "epoch": 16.0, | |
| "eval_accuracy": 0.5397766922540126, | |
| "eval_f1_macro": 0.7179414433066531, | |
| "eval_f1_micro": 0.8412167903472498, | |
| "eval_loss": 0.0919499322772026, | |
| "eval_roc_auc": 0.8980611485322447, | |
| "eval_runtime": 681.4084, | |
| "eval_samples_per_second": 4.206, | |
| "eval_steps_per_second": 0.132, | |
| "learning_rate": 0.0001, | |
| "step": 4384 | |
| }, | |
| { | |
| "epoch": 16.42, | |
| "learning_rate": 0.0001, | |
| "loss": 0.1091, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 17.0, | |
| "eval_accuracy": 0.5432658757850662, | |
| "eval_f1_macro": 0.723478025333314, | |
| "eval_f1_micro": 0.8430574400723655, | |
| "eval_loss": 0.09241489320993423, | |
| "eval_roc_auc": 0.8997269399621012, | |
| "eval_runtime": 695.7215, | |
| "eval_samples_per_second": 4.119, | |
| "eval_steps_per_second": 0.129, | |
| "learning_rate": 0.0001, | |
| "step": 4658 | |
| }, | |
| { | |
| "epoch": 18.0, | |
| "eval_accuracy": 0.5446615491974878, | |
| "eval_f1_macro": 0.7165670713509208, | |
| "eval_f1_micro": 0.8447657539705129, | |
| "eval_loss": 0.0904153436422348, | |
| "eval_roc_auc": 0.8983812561546398, | |
| "eval_runtime": 692.0734, | |
| "eval_samples_per_second": 4.141, | |
| "eval_steps_per_second": 0.13, | |
| "learning_rate": 0.0001, | |
| "step": 4932 | |
| }, | |
| { | |
| "epoch": 18.25, | |
| "learning_rate": 0.0001, | |
| "loss": 0.1026, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 19.0, | |
| "eval_accuracy": 0.5443126308443824, | |
| "eval_f1_macro": 0.7248049102675759, | |
| "eval_f1_micro": 0.8448009166427957, | |
| "eval_loss": 0.0903031975030899, | |
| "eval_roc_auc": 0.8962875391202584, | |
| "eval_runtime": 681.2634, | |
| "eval_samples_per_second": 4.207, | |
| "eval_steps_per_second": 0.132, | |
| "learning_rate": 0.0001, | |
| "step": 5206 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "eval_accuracy": 0.5422191207257502, | |
| "eval_f1_macro": 0.7175689244853282, | |
| "eval_f1_micro": 0.8439372325249643, | |
| "eval_loss": 0.08870936185121536, | |
| "eval_roc_auc": 0.897138803557627, | |
| "eval_runtime": 675.8431, | |
| "eval_samples_per_second": 4.241, | |
| "eval_steps_per_second": 0.133, | |
| "learning_rate": 0.0001, | |
| "step": 5480 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "learning_rate": 0.0001, | |
| "step": 5480, | |
| "total_flos": 2.5999409342512156e+20, | |
| "train_loss": 0.0099753277145163, | |
| "train_runtime": 5734.8315, | |
| "train_samples_per_second": 30.575, | |
| "train_steps_per_second": 0.956 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 5480, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 20, | |
| "save_steps": 500, | |
| "total_flos": 2.5999409342512156e+20, | |
| "train_batch_size": 32, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |