| { | |
| "best_metric": 0.8772656654583117, | |
| "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-cons_Classification_base_V10/checkpoint-105", | |
| "epoch": 6.885245901639344, | |
| "eval_steps": 500, | |
| "global_step": 105, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.6557377049180327, | |
| "grad_norm": 4.642137050628662, | |
| "learning_rate": 4.545454545454546e-05, | |
| "loss": 0.7975, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.9836065573770492, | |
| "step": 15, | |
| "train_accuracy": 0.8178880703114902, | |
| "train_auc_class_0": 0.8980438574304173, | |
| "train_auc_class_1": 0.9139826205363317, | |
| "train_auc_class_2": 0.8919882412630196, | |
| "train_auc_overall": 0.9013382397432562, | |
| "train_loss": 0.46714112162590027, | |
| "train_runtime": 689.8789, | |
| "train_samples_per_second": 11.215, | |
| "train_steps_per_second": 0.088 | |
| }, | |
| { | |
| "epoch": 0.9836065573770492, | |
| "eval_accuracy": 0.8524080787156914, | |
| "eval_auc_class_0": 0.9181721665274296, | |
| "eval_auc_class_1": 0.9324510717614166, | |
| "eval_auc_class_2": 0.9299151491872829, | |
| "eval_auc_overall": 0.9268461291587097, | |
| "eval_loss": 0.4173153042793274, | |
| "eval_runtime": 342.4588, | |
| "eval_samples_per_second": 5.639, | |
| "eval_steps_per_second": 0.047, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 1.3114754098360657, | |
| "grad_norm": 2.1029186248779297, | |
| "learning_rate": 4.5212765957446815e-05, | |
| "loss": 0.5023, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 1.9672131147540983, | |
| "grad_norm": 1.8800333738327026, | |
| "learning_rate": 3.9893617021276594e-05, | |
| "loss": 0.4094, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 1.9672131147540983, | |
| "step": 30, | |
| "train_accuracy": 0.8503295851105079, | |
| "train_auc_class_0": 0.9326458902508805, | |
| "train_auc_class_1": 0.9445850486238231, | |
| "train_auc_class_2": 0.9445771168575358, | |
| "train_auc_overall": 0.9406026852440799, | |
| "train_loss": 0.3614955544471741, | |
| "train_runtime": 693.803, | |
| "train_samples_per_second": 11.152, | |
| "train_steps_per_second": 0.088 | |
| }, | |
| { | |
| "epoch": 1.9672131147540983, | |
| "eval_accuracy": 0.849818746763335, | |
| "eval_auc_class_0": 0.9336946035724231, | |
| "eval_auc_class_1": 0.9488809233636819, | |
| "eval_auc_class_2": 0.9534673603715279, | |
| "eval_auc_overall": 0.9453476291025443, | |
| "eval_loss": 0.3673703968524933, | |
| "eval_runtime": 266.4299, | |
| "eval_samples_per_second": 7.248, | |
| "eval_steps_per_second": 0.06, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 2.6229508196721314, | |
| "grad_norm": 2.8551852703094482, | |
| "learning_rate": 3.4574468085106386e-05, | |
| "loss": 0.3672, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 2.9508196721311473, | |
| "step": 45, | |
| "train_accuracy": 0.864417732971436, | |
| "train_auc_class_0": 0.9455359129962017, | |
| "train_auc_class_1": 0.9545585538591823, | |
| "train_auc_class_2": 0.9593322426521459, | |
| "train_auc_overall": 0.9531422365025098, | |
| "train_loss": 0.32408276200294495, | |
| "train_runtime": 685.3567, | |
| "train_samples_per_second": 11.289, | |
| "train_steps_per_second": 0.089 | |
| }, | |
| { | |
| "epoch": 2.9508196721311473, | |
| "eval_accuracy": 0.8726048679440704, | |
| "eval_auc_class_0": 0.9429575923936826, | |
| "eval_auc_class_1": 0.9536497239945516, | |
| "eval_auc_class_2": 0.9603724007891477, | |
| "eval_auc_overall": 0.9523265723924607, | |
| "eval_loss": 0.3238150477409363, | |
| "eval_runtime": 204.7854, | |
| "eval_samples_per_second": 9.429, | |
| "eval_steps_per_second": 0.078, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 3.278688524590164, | |
| "grad_norm": 2.7802083492279053, | |
| "learning_rate": 2.925531914893617e-05, | |
| "loss": 0.3328, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 3.9344262295081966, | |
| "grad_norm": 2.6021904945373535, | |
| "learning_rate": 2.393617021276596e-05, | |
| "loss": 0.3237, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "step": 61, | |
| "train_accuracy": 0.8729481711257593, | |
| "train_auc_class_0": 0.9494105682362736, | |
| "train_auc_class_1": 0.960331196185584, | |
| "train_auc_class_2": 0.9563328693868581, | |
| "train_auc_overall": 0.955358211269572, | |
| "train_loss": 0.31062838435173035, | |
| "train_runtime": 681.8052, | |
| "train_samples_per_second": 11.348, | |
| "train_steps_per_second": 0.089 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.8720870015535992, | |
| "eval_auc_class_0": 0.9470949198392808, | |
| "eval_auc_class_1": 0.9577675819055128, | |
| "eval_auc_class_2": 0.961633814620179, | |
| "eval_auc_overall": 0.9554987721216577, | |
| "eval_loss": 0.32258161902427673, | |
| "eval_runtime": 204.0189, | |
| "eval_samples_per_second": 9.465, | |
| "eval_steps_per_second": 0.078, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 4.590163934426229, | |
| "grad_norm": 2.246328592300415, | |
| "learning_rate": 1.8617021276595745e-05, | |
| "loss": 0.3257, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 4.983606557377049, | |
| "step": 76, | |
| "train_accuracy": 0.879798371461807, | |
| "train_auc_class_0": 0.9544922319365635, | |
| "train_auc_class_1": 0.9627755801666535, | |
| "train_auc_class_2": 0.9628861811778814, | |
| "train_auc_overall": 0.9600513310936994, | |
| "train_loss": 0.29983675479888916, | |
| "train_runtime": 688.2036, | |
| "train_samples_per_second": 11.242, | |
| "train_steps_per_second": 0.089 | |
| }, | |
| { | |
| "epoch": 4.983606557377049, | |
| "eval_accuracy": 0.8741584671154842, | |
| "eval_auc_class_0": 0.9478719119226638, | |
| "eval_auc_class_1": 0.9577360384256937, | |
| "eval_auc_class_2": 0.9608001466557257, | |
| "eval_auc_overall": 0.9554693656680278, | |
| "eval_loss": 0.32330793142318726, | |
| "eval_runtime": 205.4973, | |
| "eval_samples_per_second": 9.397, | |
| "eval_steps_per_second": 0.078, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 5.245901639344262, | |
| "grad_norm": 1.7804102897644043, | |
| "learning_rate": 1.3297872340425532e-05, | |
| "loss": 0.2972, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 5.901639344262295, | |
| "grad_norm": 1.6500009298324585, | |
| "learning_rate": 7.97872340425532e-06, | |
| "loss": 0.2923, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 5.967213114754099, | |
| "step": 91, | |
| "train_accuracy": 0.8857438283572444, | |
| "train_auc_class_0": 0.9581288331197666, | |
| "train_auc_class_1": 0.966850418082771, | |
| "train_auc_class_2": 0.9694481750082071, | |
| "train_auc_overall": 0.9648091420702481, | |
| "train_loss": 0.28071513772010803, | |
| "train_runtime": 686.7084, | |
| "train_samples_per_second": 11.267, | |
| "train_steps_per_second": 0.089 | |
| }, | |
| { | |
| "epoch": 5.967213114754099, | |
| "eval_accuracy": 0.8736406007250129, | |
| "eval_auc_class_0": 0.9477737001233243, | |
| "eval_auc_class_1": 0.9590063803856907, | |
| "eval_auc_class_2": 0.9627250030553275, | |
| "eval_auc_overall": 0.9565016945214474, | |
| "eval_loss": 0.3128957450389862, | |
| "eval_runtime": 206.0937, | |
| "eval_samples_per_second": 9.37, | |
| "eval_steps_per_second": 0.078, | |
| "step": 91 | |
| }, | |
| { | |
| "epoch": 6.557377049180328, | |
| "grad_norm": 2.4247124195098877, | |
| "learning_rate": 2.6595744680851065e-06, | |
| "loss": 0.2794, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 6.885245901639344, | |
| "step": 105, | |
| "train_accuracy": 0.8973762440222308, | |
| "train_auc_class_0": 0.9638374809913217, | |
| "train_auc_class_1": 0.9706794670135619, | |
| "train_auc_class_2": 0.9722462334871682, | |
| "train_auc_overall": 0.9689210604973506, | |
| "train_loss": 0.26410263776779175, | |
| "train_runtime": 684.8081, | |
| "train_samples_per_second": 11.298, | |
| "train_steps_per_second": 0.089 | |
| }, | |
| { | |
| "epoch": 6.885245901639344, | |
| "eval_accuracy": 0.8772656654583117, | |
| "eval_auc_class_0": 0.9495477284481044, | |
| "eval_auc_class_1": 0.9596716610509716, | |
| "eval_auc_class_2": 0.9626115194580721, | |
| "eval_auc_overall": 0.9572769696523826, | |
| "eval_loss": 0.30786964297294617, | |
| "eval_runtime": 205.1702, | |
| "eval_samples_per_second": 9.412, | |
| "eval_steps_per_second": 0.078, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 6.885245901639344, | |
| "step": 105, | |
| "total_flos": 4.1785312376666235e+18, | |
| "train_loss": 0.38806929134187246, | |
| "train_runtime": 12116.6555, | |
| "train_samples_per_second": 4.47, | |
| "train_steps_per_second": 0.009 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 105, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 7, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 4.1785312376666235e+18, | |
| "train_batch_size": 128, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |