|
{ |
|
"best_metric": 0.5376309156417847, |
|
"best_model_checkpoint": "./plant-classification/checkpoint-160", |
|
"epoch": 2.0, |
|
"global_step": 160, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 0.0001875, |
|
"loss": 2.4258, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 0.000175, |
|
"loss": 1.9889, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 0.00016250000000000002, |
|
"loss": 1.607, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 0.00015000000000000001, |
|
"loss": 1.3676, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"eval_accuracy": 0.7850467289719626, |
|
"eval_loss": 1.1707247495651245, |
|
"eval_runtime": 340.2579, |
|
"eval_samples_per_second": 1.572, |
|
"eval_steps_per_second": 0.197, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 0.0001375, |
|
"loss": 1.1479, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 0.000125, |
|
"loss": 1.041, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 0.00011250000000000001, |
|
"loss": 0.9377, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 0.0001, |
|
"loss": 0.7748, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_accuracy": 0.8242990654205608, |
|
"eval_loss": 0.7840399146080017, |
|
"eval_runtime": 345.814, |
|
"eval_samples_per_second": 1.547, |
|
"eval_steps_per_second": 0.194, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 1.12, |
|
"learning_rate": 8.75e-05, |
|
"loss": 0.6445, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 1.25, |
|
"learning_rate": 7.500000000000001e-05, |
|
"loss": 0.6609, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 1.38, |
|
"learning_rate": 6.25e-05, |
|
"loss": 0.5166, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 1.5, |
|
"learning_rate": 5e-05, |
|
"loss": 0.5183, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 1.5, |
|
"eval_accuracy": 0.8467289719626169, |
|
"eval_loss": 0.5941035747528076, |
|
"eval_runtime": 349.1506, |
|
"eval_samples_per_second": 1.532, |
|
"eval_steps_per_second": 0.192, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 1.62, |
|
"learning_rate": 3.7500000000000003e-05, |
|
"loss": 0.4848, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 1.75, |
|
"learning_rate": 2.5e-05, |
|
"loss": 0.5598, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 1.88, |
|
"learning_rate": 1.25e-05, |
|
"loss": 0.4853, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 0.0, |
|
"loss": 0.4338, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy": 0.8822429906542056, |
|
"eval_loss": 0.5376309156417847, |
|
"eval_runtime": 365.9435, |
|
"eval_samples_per_second": 1.462, |
|
"eval_steps_per_second": 0.183, |
|
"step": 160 |
|
} |
|
], |
|
"max_steps": 160, |
|
"num_train_epochs": 2, |
|
"total_flos": 1.9809082826897818e+17, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|