{ "best_metric": 0.0010297322878614068, "best_model_checkpoint": "./vit-base-modified-augmented-ph2-patch-16/checkpoint-650", "epoch": 4.0, "global_step": 680, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.06, "learning_rate": 0.00019705882352941177, "loss": 0.4087, "step": 10 }, { "epoch": 0.12, "learning_rate": 0.00019411764705882354, "loss": 0.2268, "step": 20 }, { "epoch": 0.18, "learning_rate": 0.0001911764705882353, "loss": 0.1001, "step": 30 }, { "epoch": 0.24, "learning_rate": 0.00018823529411764707, "loss": 0.0906, "step": 40 }, { "epoch": 0.29, "learning_rate": 0.00018529411764705883, "loss": 0.1238, "step": 50 }, { "epoch": 0.29, "eval_accuracy": 0.9332191780821918, "eval_f1": 0.9332191780821918, "eval_loss": 0.1972886323928833, "eval_precision": 0.9332191780821918, "eval_recall": 0.9332191780821918, "eval_runtime": 10.5543, "eval_samples_per_second": 55.333, "eval_steps_per_second": 6.917, "step": 50 }, { "epoch": 0.35, "learning_rate": 0.0001823529411764706, "loss": 0.0567, "step": 60 }, { "epoch": 0.41, "learning_rate": 0.00017941176470588236, "loss": 0.1442, "step": 70 }, { "epoch": 0.47, "learning_rate": 0.00017647058823529413, "loss": 0.0469, "step": 80 }, { "epoch": 0.53, "learning_rate": 0.0001735294117647059, "loss": 0.0421, "step": 90 }, { "epoch": 0.59, "learning_rate": 0.00017058823529411766, "loss": 0.1857, "step": 100 }, { "epoch": 0.59, "eval_accuracy": 0.9623287671232876, "eval_f1": 0.9623287671232876, "eval_loss": 0.10843866318464279, "eval_precision": 0.9623287671232876, "eval_recall": 0.9623287671232876, "eval_runtime": 10.5889, "eval_samples_per_second": 55.152, "eval_steps_per_second": 6.894, "step": 100 }, { "epoch": 0.65, "learning_rate": 0.00016764705882352942, "loss": 0.0418, "step": 110 }, { "epoch": 0.71, "learning_rate": 0.0001647058823529412, "loss": 0.0318, "step": 120 }, { "epoch": 0.76, "learning_rate": 0.00016176470588235295, "loss": 0.0121, "step": 130 }, { "epoch": 0.82, "learning_rate": 0.0001588235294117647, "loss": 0.034, "step": 140 }, { "epoch": 0.88, "learning_rate": 0.00015588235294117648, "loss": 0.2506, "step": 150 }, { "epoch": 0.88, "eval_accuracy": 0.9691780821917808, "eval_f1": 0.9691780821917808, "eval_loss": 0.0773317888379097, "eval_precision": 0.9691780821917808, "eval_recall": 0.9691780821917808, "eval_runtime": 10.5182, "eval_samples_per_second": 55.523, "eval_steps_per_second": 6.94, "step": 150 }, { "epoch": 0.94, "learning_rate": 0.00015294117647058822, "loss": 0.0931, "step": 160 }, { "epoch": 1.0, "learning_rate": 0.00015000000000000001, "loss": 0.0237, "step": 170 }, { "epoch": 1.06, "learning_rate": 0.00014705882352941178, "loss": 0.0549, "step": 180 }, { "epoch": 1.12, "learning_rate": 0.00014411764705882354, "loss": 0.0089, "step": 190 }, { "epoch": 1.18, "learning_rate": 0.0001411764705882353, "loss": 0.0247, "step": 200 }, { "epoch": 1.18, "eval_accuracy": 0.9606164383561644, "eval_f1": 0.9606164383561644, "eval_loss": 0.11584462225437164, "eval_precision": 0.9606164383561644, "eval_recall": 0.9606164383561644, "eval_runtime": 10.5145, "eval_samples_per_second": 55.543, "eval_steps_per_second": 6.943, "step": 200 }, { "epoch": 1.24, "learning_rate": 0.00013823529411764707, "loss": 0.0037, "step": 210 }, { "epoch": 1.29, "learning_rate": 0.00013529411764705884, "loss": 0.0034, "step": 220 }, { "epoch": 1.35, "learning_rate": 0.0001323529411764706, "loss": 0.0032, "step": 230 }, { "epoch": 1.41, "learning_rate": 0.00012941176470588237, "loss": 0.0029, "step": 240 }, { "epoch": 1.47, "learning_rate": 0.0001264705882352941, "loss": 0.0089, "step": 250 }, { "epoch": 1.47, "eval_accuracy": 0.9914383561643836, "eval_f1": 0.9914383561643836, "eval_loss": 0.01619771681725979, "eval_precision": 0.9914383561643836, "eval_recall": 0.9914383561643836, "eval_runtime": 10.5615, "eval_samples_per_second": 55.295, "eval_steps_per_second": 6.912, "step": 250 }, { "epoch": 1.53, "learning_rate": 0.0001235294117647059, "loss": 0.0084, "step": 260 }, { "epoch": 1.59, "learning_rate": 0.00012058823529411765, "loss": 0.0234, "step": 270 }, { "epoch": 1.65, "learning_rate": 0.00011764705882352942, "loss": 0.0023, "step": 280 }, { "epoch": 1.71, "learning_rate": 0.00011470588235294118, "loss": 0.0021, "step": 290 }, { "epoch": 1.76, "learning_rate": 0.00011176470588235294, "loss": 0.0226, "step": 300 }, { "epoch": 1.76, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.001967146061360836, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.8533, "eval_samples_per_second": 53.809, "eval_steps_per_second": 6.726, "step": 300 }, { "epoch": 1.82, "learning_rate": 0.0001088235294117647, "loss": 0.0022, "step": 310 }, { "epoch": 1.88, "learning_rate": 0.00010588235294117647, "loss": 0.0025, "step": 320 }, { "epoch": 1.94, "learning_rate": 0.00010294117647058823, "loss": 0.0017, "step": 330 }, { "epoch": 2.0, "learning_rate": 0.0001, "loss": 0.0016, "step": 340 }, { "epoch": 2.06, "learning_rate": 9.705882352941177e-05, "loss": 0.0261, "step": 350 }, { "epoch": 2.06, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.0017063928535208106, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.5279, "eval_samples_per_second": 55.472, "eval_steps_per_second": 6.934, "step": 350 }, { "epoch": 2.12, "learning_rate": 9.411764705882353e-05, "loss": 0.0017, "step": 360 }, { "epoch": 2.18, "learning_rate": 9.11764705882353e-05, "loss": 0.0063, "step": 370 }, { "epoch": 2.24, "learning_rate": 8.823529411764706e-05, "loss": 0.002, "step": 380 }, { "epoch": 2.29, "learning_rate": 8.529411764705883e-05, "loss": 0.0129, "step": 390 }, { "epoch": 2.35, "learning_rate": 8.23529411764706e-05, "loss": 0.0014, "step": 400 }, { "epoch": 2.35, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.0013761176960542798, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.7957, "eval_samples_per_second": 54.096, "eval_steps_per_second": 6.762, "step": 400 }, { "epoch": 2.41, "learning_rate": 7.941176470588235e-05, "loss": 0.0014, "step": 410 }, { "epoch": 2.47, "learning_rate": 7.647058823529411e-05, "loss": 0.0326, "step": 420 }, { "epoch": 2.53, "learning_rate": 7.352941176470589e-05, "loss": 0.0044, "step": 430 }, { "epoch": 2.59, "learning_rate": 7.058823529411765e-05, "loss": 0.0013, "step": 440 }, { "epoch": 2.65, "learning_rate": 6.764705882352942e-05, "loss": 0.0012, "step": 450 }, { "epoch": 2.65, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.0012688947608694434, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.501, "eval_samples_per_second": 55.614, "eval_steps_per_second": 6.952, "step": 450 }, { "epoch": 2.71, "learning_rate": 6.470588235294118e-05, "loss": 0.0012, "step": 460 }, { "epoch": 2.76, "learning_rate": 6.176470588235295e-05, "loss": 0.0012, "step": 470 }, { "epoch": 2.82, "learning_rate": 5.882352941176471e-05, "loss": 0.0234, "step": 480 }, { "epoch": 2.88, "learning_rate": 5.588235294117647e-05, "loss": 0.0379, "step": 490 }, { "epoch": 2.94, "learning_rate": 5.294117647058824e-05, "loss": 0.0013, "step": 500 }, { "epoch": 2.94, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.00120103033259511, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.468, "eval_samples_per_second": 55.789, "eval_steps_per_second": 6.974, "step": 500 }, { "epoch": 3.0, "learning_rate": 5e-05, "loss": 0.0012, "step": 510 }, { "epoch": 3.06, "learning_rate": 4.705882352941177e-05, "loss": 0.0012, "step": 520 }, { "epoch": 3.12, "learning_rate": 4.411764705882353e-05, "loss": 0.0012, "step": 530 }, { "epoch": 3.18, "learning_rate": 4.11764705882353e-05, "loss": 0.0011, "step": 540 }, { "epoch": 3.24, "learning_rate": 3.8235294117647055e-05, "loss": 0.0011, "step": 550 }, { "epoch": 3.24, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.0011208122596144676, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.4395, "eval_samples_per_second": 55.941, "eval_steps_per_second": 6.993, "step": 550 }, { "epoch": 3.29, "learning_rate": 3.529411764705883e-05, "loss": 0.0011, "step": 560 }, { "epoch": 3.35, "learning_rate": 3.235294117647059e-05, "loss": 0.0011, "step": 570 }, { "epoch": 3.41, "learning_rate": 2.9411764705882354e-05, "loss": 0.0011, "step": 580 }, { "epoch": 3.47, "learning_rate": 2.647058823529412e-05, "loss": 0.001, "step": 590 }, { "epoch": 3.53, "learning_rate": 2.3529411764705884e-05, "loss": 0.001, "step": 600 }, { "epoch": 3.53, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.0010598464868962765, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.4617, "eval_samples_per_second": 55.823, "eval_steps_per_second": 6.978, "step": 600 }, { "epoch": 3.59, "learning_rate": 2.058823529411765e-05, "loss": 0.001, "step": 610 }, { "epoch": 3.65, "learning_rate": 1.7647058823529414e-05, "loss": 0.001, "step": 620 }, { "epoch": 3.71, "learning_rate": 1.4705882352941177e-05, "loss": 0.001, "step": 630 }, { "epoch": 3.76, "learning_rate": 1.1764705882352942e-05, "loss": 0.001, "step": 640 }, { "epoch": 3.82, "learning_rate": 8.823529411764707e-06, "loss": 0.0011, "step": 650 }, { "epoch": 3.82, "eval_accuracy": 1.0, "eval_f1": 1.0, "eval_loss": 0.0010297322878614068, "eval_precision": 1.0, "eval_recall": 1.0, "eval_runtime": 10.5136, "eval_samples_per_second": 55.547, "eval_steps_per_second": 6.943, "step": 650 }, { "epoch": 3.88, "learning_rate": 5.882352941176471e-06, "loss": 0.001, "step": 660 }, { "epoch": 3.94, "learning_rate": 2.9411764705882355e-06, "loss": 0.001, "step": 670 }, { "epoch": 4.0, "learning_rate": 0.0, "loss": 0.0011, "step": 680 }, { "epoch": 4.0, "step": 680, "total_flos": 8.412530392560108e+17, "train_loss": 0.03332017799154582, "train_runtime": 491.3331, "train_samples_per_second": 22.095, "train_steps_per_second": 1.384 } ], "max_steps": 680, "num_train_epochs": 4, "total_flos": 8.412530392560108e+17, "trial_name": null, "trial_params": null }