{ "best_metric": 0.9651567944250871, "best_model_checkpoint": "brain-tumor-classification/checkpoint-200", "epoch": 4.938271604938271, "eval_steps": 500, "global_step": 200, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.24691358024691357, "grad_norm": 9.583415985107422, "learning_rate": 2.5e-05, "loss": 0.0798, "step": 10 }, { "epoch": 0.49382716049382713, "grad_norm": 4.763514518737793, "learning_rate": 5e-05, "loss": 0.0516, "step": 20 }, { "epoch": 0.7407407407407407, "grad_norm": 3.712924003601074, "learning_rate": 4.722222222222222e-05, "loss": 0.0708, "step": 30 }, { "epoch": 0.9876543209876543, "grad_norm": 3.5341458320617676, "learning_rate": 4.4444444444444447e-05, "loss": 0.0628, "step": 40 }, { "epoch": 0.9876543209876543, "eval_accuracy": 0.9512195121951219, "eval_loss": 0.14935272932052612, "eval_runtime": 2.3845, "eval_samples_per_second": 120.36, "eval_steps_per_second": 7.549, "step": 40 }, { "epoch": 1.2345679012345678, "grad_norm": 2.3918871879577637, "learning_rate": 4.166666666666667e-05, "loss": 0.0562, "step": 50 }, { "epoch": 1.4814814814814814, "grad_norm": 7.054160118103027, "learning_rate": 3.888888888888889e-05, "loss": 0.0587, "step": 60 }, { "epoch": 1.7283950617283952, "grad_norm": 16.58973503112793, "learning_rate": 3.611111111111111e-05, "loss": 0.0679, "step": 70 }, { "epoch": 1.9753086419753085, "grad_norm": 11.970051765441895, "learning_rate": 3.3333333333333335e-05, "loss": 0.0499, "step": 80 }, { "epoch": 2.0, "eval_accuracy": 0.9372822299651568, "eval_loss": 0.18301770091056824, "eval_runtime": 2.3848, "eval_samples_per_second": 120.346, "eval_steps_per_second": 7.548, "step": 81 }, { "epoch": 2.2222222222222223, "grad_norm": 6.739134788513184, "learning_rate": 3.055555555555556e-05, "loss": 0.0352, "step": 90 }, { "epoch": 2.4691358024691357, "grad_norm": 23.700721740722656, "learning_rate": 2.777777777777778e-05, "loss": 0.1, "step": 100 }, { "epoch": 2.7160493827160495, "grad_norm": 6.7289581298828125, "learning_rate": 2.5e-05, "loss": 0.0504, "step": 110 }, { "epoch": 2.962962962962963, "grad_norm": 11.606633186340332, "learning_rate": 2.2222222222222223e-05, "loss": 0.0697, "step": 120 }, { "epoch": 2.9876543209876543, "eval_accuracy": 0.9547038327526133, "eval_loss": 0.10937974601984024, "eval_runtime": 2.9179, "eval_samples_per_second": 98.36, "eval_steps_per_second": 6.169, "step": 121 }, { "epoch": 3.2098765432098766, "grad_norm": 17.90212059020996, "learning_rate": 1.9444444444444445e-05, "loss": 0.0546, "step": 130 }, { "epoch": 3.45679012345679, "grad_norm": 8.192621231079102, "learning_rate": 1.6666666666666667e-05, "loss": 0.0601, "step": 140 }, { "epoch": 3.7037037037037037, "grad_norm": 3.0072553157806396, "learning_rate": 1.388888888888889e-05, "loss": 0.07, "step": 150 }, { "epoch": 3.950617283950617, "grad_norm": 5.344573974609375, "learning_rate": 1.1111111111111112e-05, "loss": 0.0646, "step": 160 }, { "epoch": 4.0, "eval_accuracy": 0.9581881533101045, "eval_loss": 0.13176245987415314, "eval_runtime": 2.8306, "eval_samples_per_second": 101.391, "eval_steps_per_second": 6.359, "step": 162 }, { "epoch": 4.197530864197531, "grad_norm": 12.885249137878418, "learning_rate": 8.333333333333334e-06, "loss": 0.0515, "step": 170 }, { "epoch": 4.444444444444445, "grad_norm": 13.122868537902832, "learning_rate": 5.555555555555556e-06, "loss": 0.0784, "step": 180 }, { "epoch": 4.6913580246913575, "grad_norm": 8.814871788024902, "learning_rate": 2.777777777777778e-06, "loss": 0.088, "step": 190 }, { "epoch": 4.938271604938271, "grad_norm": 1.9182705879211426, "learning_rate": 0.0, "loss": 0.0481, "step": 200 }, { "epoch": 4.938271604938271, "eval_accuracy": 0.9651567944250871, "eval_loss": 0.11668162047863007, "eval_runtime": 2.6735, "eval_samples_per_second": 107.351, "eval_steps_per_second": 6.733, "step": 200 }, { "epoch": 4.938271604938271, "step": 200, "total_flos": 3.1727957353537536e+17, "train_loss": 0.06341908916831017, "train_runtime": 245.9692, "train_samples_per_second": 52.507, "train_steps_per_second": 0.813 } ], "logging_steps": 10, "max_steps": 200, "num_input_tokens_seen": 0, "num_train_epochs": 5, "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": 3.1727957353537536e+17, "train_batch_size": 16, "trial_name": null, "trial_params": null }