{ "best_metric": 0.8990825688073395, "best_model_checkpoint": "google-vit-base-patch16-224-cartoon-emotion-detection/checkpoint-56", "epoch": 9.969696969696969, "global_step": 80, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.97, "eval_accuracy": 0.5596330275229358, "eval_f1": 0.5360425549075453, "eval_loss": 0.9901540279388428, "eval_precision": 0.5505540964988554, "eval_recall": 0.5596330275229358, "eval_runtime": 5.4111, "eval_samples_per_second": 20.144, "eval_steps_per_second": 0.37, "step": 8 }, { "epoch": 1.24, "learning_rate": 0.00011666666666666667, "loss": 1.242, "step": 10 }, { "epoch": 1.97, "eval_accuracy": 0.8165137614678899, "eval_f1": 0.8131755577433994, "eval_loss": 0.5157326459884644, "eval_precision": 0.819457058574903, "eval_recall": 0.8165137614678899, "eval_runtime": 4.6435, "eval_samples_per_second": 23.473, "eval_steps_per_second": 0.431, "step": 16 }, { "epoch": 2.48, "learning_rate": 0.0001, "loss": 0.4438, "step": 20 }, { "epoch": 2.97, "eval_accuracy": 0.8440366972477065, "eval_f1": 0.8446079913969823, "eval_loss": 0.3870726525783539, "eval_precision": 0.8516382699868937, "eval_recall": 0.8440366972477065, "eval_runtime": 4.4313, "eval_samples_per_second": 24.598, "eval_steps_per_second": 0.451, "step": 24 }, { "epoch": 3.73, "learning_rate": 8.333333333333333e-05, "loss": 0.1768, "step": 30 }, { "epoch": 3.97, "eval_accuracy": 0.8623853211009175, "eval_f1": 0.8585073146382761, "eval_loss": 0.353137731552124, "eval_precision": 0.865271038673443, "eval_recall": 0.8623853211009175, "eval_runtime": 6.7356, "eval_samples_per_second": 16.183, "eval_steps_per_second": 0.297, "step": 32 }, { "epoch": 4.97, "learning_rate": 6.666666666666667e-05, "loss": 0.0661, "step": 40 }, { "epoch": 4.97, "eval_accuracy": 0.8715596330275229, "eval_f1": 0.8673695820239424, "eval_loss": 0.3780345022678375, "eval_precision": 0.8693191511585397, "eval_recall": 0.8715596330275229, "eval_runtime": 4.3244, "eval_samples_per_second": 25.206, "eval_steps_per_second": 0.462, "step": 40 }, { "epoch": 5.97, "eval_accuracy": 0.8623853211009175, "eval_f1": 0.8631844663686221, "eval_loss": 0.374713659286499, "eval_precision": 0.864939209175348, "eval_recall": 0.8623853211009175, "eval_runtime": 4.479, "eval_samples_per_second": 24.336, "eval_steps_per_second": 0.447, "step": 48 }, { "epoch": 6.24, "learning_rate": 5e-05, "loss": 0.0375, "step": 50 }, { "epoch": 6.97, "eval_accuracy": 0.8990825688073395, "eval_f1": 0.8970800861485594, "eval_loss": 0.37597015500068665, "eval_precision": 0.8961258037611989, "eval_recall": 0.8990825688073395, "eval_runtime": 4.4765, "eval_samples_per_second": 24.349, "eval_steps_per_second": 0.447, "step": 56 }, { "epoch": 7.48, "learning_rate": 3.3333333333333335e-05, "loss": 0.0362, "step": 60 }, { "epoch": 7.97, "eval_accuracy": 0.8715596330275229, "eval_f1": 0.8680866720523425, "eval_loss": 0.4092080295085907, "eval_precision": 0.8684143462632116, "eval_recall": 0.8715596330275229, "eval_runtime": 4.3876, "eval_samples_per_second": 24.843, "eval_steps_per_second": 0.456, "step": 64 }, { "epoch": 8.73, "learning_rate": 1.6666666666666667e-05, "loss": 0.0322, "step": 70 }, { "epoch": 8.97, "eval_accuracy": 0.8899082568807339, "eval_f1": 0.8888099976565806, "eval_loss": 0.3499366044998169, "eval_precision": 0.8880198424442283, "eval_recall": 0.8899082568807339, "eval_runtime": 5.3381, "eval_samples_per_second": 20.419, "eval_steps_per_second": 0.375, "step": 72 }, { "epoch": 9.97, "learning_rate": 0.0, "loss": 0.029, "step": 80 }, { "epoch": 9.97, "eval_accuracy": 0.8807339449541285, "eval_f1": 0.8782945902988435, "eval_loss": 0.37059423327445984, "eval_precision": 0.8768597487153273, "eval_recall": 0.8807339449541285, "eval_runtime": 4.4763, "eval_samples_per_second": 24.35, "eval_steps_per_second": 0.447, "step": 80 }, { "epoch": 9.97, "step": 80, "total_flos": 1.6024080844892897e+18, "train_loss": 0.2579587835818529, "train_runtime": 415.4218, "train_samples_per_second": 49.829, "train_steps_per_second": 0.193 } ], "max_steps": 80, "num_train_epochs": 10, "total_flos": 1.6024080844892897e+18, "trial_name": null, "trial_params": null }