| { | |
| "best_metric": 0.852151095867157, | |
| "best_model_checkpoint": "./vit-Arousal/checkpoint-100", | |
| "epoch": 4.0, | |
| "eval_steps": 100, | |
| "global_step": 512, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 0.00019609375, | |
| "loss": 1.3074, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 0.0001921875, | |
| "loss": 1.0019, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 0.00018828125, | |
| "loss": 0.9923, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 0.000184375, | |
| "loss": 0.8658, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 0.00018046875000000002, | |
| "loss": 0.9941, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 0.00017656250000000002, | |
| "loss": 1.0082, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 0.00017265625, | |
| "loss": 0.949, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "learning_rate": 0.00016875, | |
| "loss": 0.9459, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 0.00016484375, | |
| "loss": 0.8496, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "learning_rate": 0.0001609375, | |
| "loss": 0.9023, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "eval_accuracy": 0.6294117647058823, | |
| "eval_loss": 0.852151095867157, | |
| "eval_runtime": 9.2104, | |
| "eval_samples_per_second": 55.372, | |
| "eval_steps_per_second": 6.949, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "learning_rate": 0.00015703125, | |
| "loss": 0.8067, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "learning_rate": 0.000153125, | |
| "loss": 0.8751, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 1.02, | |
| "learning_rate": 0.00014921875000000002, | |
| "loss": 0.7552, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.09, | |
| "learning_rate": 0.00014531250000000002, | |
| "loss": 0.5661, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.17, | |
| "learning_rate": 0.00014140625, | |
| "loss": 0.549, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "learning_rate": 0.0001375, | |
| "loss": 0.6449, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "learning_rate": 0.00013359375, | |
| "loss": 0.5941, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.41, | |
| "learning_rate": 0.0001296875, | |
| "loss": 0.5656, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.48, | |
| "learning_rate": 0.00012578125, | |
| "loss": 0.542, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.56, | |
| "learning_rate": 0.00012187500000000001, | |
| "loss": 0.5376, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.56, | |
| "eval_accuracy": 0.6686274509803921, | |
| "eval_loss": 0.8591523766517639, | |
| "eval_runtime": 8.6666, | |
| "eval_samples_per_second": 58.846, | |
| "eval_steps_per_second": 7.385, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.64, | |
| "learning_rate": 0.00011796875, | |
| "loss": 0.6298, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.72, | |
| "learning_rate": 0.0001140625, | |
| "loss": 0.5563, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.8, | |
| "learning_rate": 0.00011015625000000001, | |
| "loss": 0.6173, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.88, | |
| "learning_rate": 0.00010625000000000001, | |
| "loss": 0.6663, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.95, | |
| "learning_rate": 0.00010234375, | |
| "loss": 0.5781, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 2.03, | |
| "learning_rate": 9.84375e-05, | |
| "loss": 0.528, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 2.11, | |
| "learning_rate": 9.453125000000001e-05, | |
| "loss": 0.2836, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 2.19, | |
| "learning_rate": 9.062500000000001e-05, | |
| "loss": 0.2782, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 2.27, | |
| "learning_rate": 8.671875e-05, | |
| "loss": 0.2749, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 2.34, | |
| "learning_rate": 8.28125e-05, | |
| "loss": 0.2473, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.34, | |
| "eval_accuracy": 0.6509803921568628, | |
| "eval_loss": 0.9559371471405029, | |
| "eval_runtime": 9.3899, | |
| "eval_samples_per_second": 54.314, | |
| "eval_steps_per_second": 6.816, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.42, | |
| "learning_rate": 7.890625000000001e-05, | |
| "loss": 0.1479, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 2.5, | |
| "learning_rate": 7.500000000000001e-05, | |
| "loss": 0.3088, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 2.58, | |
| "learning_rate": 7.109375e-05, | |
| "loss": 0.242, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 2.66, | |
| "learning_rate": 6.71875e-05, | |
| "loss": 0.2083, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 2.73, | |
| "learning_rate": 6.328125e-05, | |
| "loss": 0.2094, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.81, | |
| "learning_rate": 5.9375e-05, | |
| "loss": 0.1957, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 2.89, | |
| "learning_rate": 5.546875e-05, | |
| "loss": 0.1911, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "learning_rate": 5.15625e-05, | |
| "loss": 0.2174, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 3.05, | |
| "learning_rate": 4.765625e-05, | |
| "loss": 0.1176, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 3.12, | |
| "learning_rate": 4.375e-05, | |
| "loss": 0.0691, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 3.12, | |
| "eval_accuracy": 0.6274509803921569, | |
| "eval_loss": 1.1399476528167725, | |
| "eval_runtime": 8.471, | |
| "eval_samples_per_second": 60.206, | |
| "eval_steps_per_second": 7.555, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 3.2, | |
| "learning_rate": 3.984375e-05, | |
| "loss": 0.0712, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 3.28, | |
| "learning_rate": 3.59375e-05, | |
| "loss": 0.0492, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 3.36, | |
| "learning_rate": 3.203125e-05, | |
| "loss": 0.0818, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 3.44, | |
| "learning_rate": 2.8125000000000003e-05, | |
| "loss": 0.0854, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 3.52, | |
| "learning_rate": 2.4218750000000003e-05, | |
| "loss": 0.1115, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 3.59, | |
| "learning_rate": 2.0312500000000002e-05, | |
| "loss": 0.1219, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 3.67, | |
| "learning_rate": 1.6406250000000002e-05, | |
| "loss": 0.0581, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 3.75, | |
| "learning_rate": 1.25e-05, | |
| "loss": 0.0745, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 3.83, | |
| "learning_rate": 8.59375e-06, | |
| "loss": 0.039, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 3.91, | |
| "learning_rate": 4.6875000000000004e-06, | |
| "loss": 0.0821, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.91, | |
| "eval_accuracy": 0.6392156862745098, | |
| "eval_loss": 1.20601224899292, | |
| "eval_runtime": 9.3464, | |
| "eval_samples_per_second": 54.566, | |
| "eval_steps_per_second": 6.848, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.98, | |
| "learning_rate": 7.8125e-07, | |
| "loss": 0.06, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "step": 512, | |
| "total_flos": 6.311117323294802e+17, | |
| "train_loss": 0.46213533167610876, | |
| "train_runtime": 403.5009, | |
| "train_samples_per_second": 20.183, | |
| "train_steps_per_second": 1.269 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 512, | |
| "num_train_epochs": 4, | |
| "save_steps": 100, | |
| "total_flos": 6.311117323294802e+17, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |