{ "best_metric": 0.992, "best_model_checkpoint": "2025-01-21-15-57-43-swin-base-patch4-window7-224/checkpoint-102", "epoch": 5.0, "eval_steps": 500, "global_step": 170, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "train_accuracy": 0.7185185185185186 }, { "epoch": 1.0, "grad_norm": 23.64792823791504, "learning_rate": 0.0002, "loss": 0.732, "step": 34 }, { "epoch": 1.0, "eval_accuracy": 0.864857142857143, "eval_error_rate": 0.135142857142857, "eval_f1": 0.8590001766472355, "eval_loss": 0.3980206549167633, "eval_precision": 0.9164785420340976, "eval_recall": 0.8740740740740741, "eval_runtime": 3.5078, "eval_samples_per_second": 38.486, "eval_steps_per_second": 1.425, "eval_top1_accuracy": 0.8740740740740741, "step": 34 }, { "epoch": 2.0, "train_accuracy": 0.9185185185185185 }, { "epoch": 2.0, "grad_norm": 8.839152336120605, "learning_rate": 0.00017777777777777779, "loss": 0.2462, "step": 68 }, { "epoch": 2.0, "eval_accuracy": 0.9499066705963257, "eval_error_rate": 0.05009332940367428, "eval_f1": 0.948411944708241, "eval_loss": 0.10511945188045502, "eval_precision": 0.9538222952016054, "eval_recall": 0.9481481481481482, "eval_runtime": 3.5401, "eval_samples_per_second": 38.135, "eval_steps_per_second": 1.412, "eval_top1_accuracy": 0.9481481481481482, "step": 68 }, { "epoch": 3.0, "train_accuracy": 0.9349794238683128 }, { "epoch": 3.0, "grad_norm": 8.658217430114746, "learning_rate": 0.00015555555555555556, "loss": 0.1991, "step": 102 }, { "epoch": 3.0, "eval_accuracy": 0.992, "eval_error_rate": 0.008000000000000007, "eval_f1": 0.9925797814417668, "eval_loss": 0.03836945444345474, "eval_precision": 0.9928395061728394, "eval_recall": 0.9925925925925926, "eval_runtime": 3.3904, "eval_samples_per_second": 39.819, "eval_steps_per_second": 1.475, "eval_top1_accuracy": 0.9925925925925926, "step": 102 }, { "epoch": 4.0, "train_accuracy": 0.9530864197530864 }, { "epoch": 4.0, "grad_norm": 7.14619779586792, "learning_rate": 0.00013333333333333334, "loss": 0.1559, "step": 136 }, { "epoch": 4.0, "eval_accuracy": 0.9777188328912467, "eval_error_rate": 0.022281167108753275, "eval_f1": 0.9780454350411318, "eval_loss": 0.08900710195302963, "eval_precision": 0.98015873015873, "eval_recall": 0.9777777777777777, "eval_runtime": 3.432, "eval_samples_per_second": 39.335, "eval_steps_per_second": 1.457, "eval_top1_accuracy": 0.9777777777777777, "step": 136 }, { "epoch": 5.0, "train_accuracy": 0.9703703703703703 }, { "epoch": 5.0, "grad_norm": 5.705554008483887, "learning_rate": 0.00011111111111111112, "loss": 0.1024, "step": 170 }, { "epoch": 5.0, "eval_accuracy": 0.9846153846153847, "eval_error_rate": 0.01538461538461533, "eval_f1": 0.9851737891737893, "eval_loss": 0.10920833796262741, "eval_precision": 0.9862825788751715, "eval_recall": 0.9851851851851852, "eval_runtime": 3.7553, "eval_samples_per_second": 35.949, "eval_steps_per_second": 1.331, "eval_top1_accuracy": 0.9851851851851852, "step": 170 }, { "epoch": 5.0, "step": 170, "total_flos": 4.230809974960128e+17, "train_loss": 0.2870972801657284, "train_runtime": 294.683, "train_samples_per_second": 36.65, "train_steps_per_second": 1.154 } ], "logging_steps": 1, "max_steps": 340, "num_input_tokens_seen": 0, "num_train_epochs": 10, "save_steps": 500, "stateful_callbacks": { "EarlyStoppingCallback": { "args": { "early_stopping_patience": 2, "early_stopping_threshold": 0.0 }, "attributes": { "early_stopping_patience_counter": 2 } }, "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 4.230809974960128e+17, "train_batch_size": 32, "trial_name": null, "trial_params": null }