resnet-50-drfx-surgery-classifier / trainer_state.json
jayanthspratap's picture
End of training
6719fbb
{
"best_metric": 0.875,
"best_model_checkpoint": "resnet-50-drfx-surgery-classifier/checkpoint-8",
"epoch": 10.0,
"global_step": 40,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"eval_accuracy": 0.8125,
"eval_loss": 0.6591495871543884,
"eval_runtime": 0.1233,
"eval_samples_per_second": 129.795,
"eval_steps_per_second": 32.449,
"step": 4
},
{
"epoch": 2.0,
"eval_accuracy": 0.875,
"eval_loss": 0.6399165391921997,
"eval_runtime": 0.303,
"eval_samples_per_second": 52.801,
"eval_steps_per_second": 13.2,
"step": 8
},
{
"epoch": 2.5,
"learning_rate": 4.166666666666667e-06,
"loss": 0.6638,
"step": 10
},
{
"epoch": 3.0,
"eval_accuracy": 0.875,
"eval_loss": 0.6670691967010498,
"eval_runtime": 0.1445,
"eval_samples_per_second": 110.718,
"eval_steps_per_second": 27.68,
"step": 12
},
{
"epoch": 4.0,
"eval_accuracy": 0.8125,
"eval_loss": 0.6644818782806396,
"eval_runtime": 0.1229,
"eval_samples_per_second": 130.239,
"eval_steps_per_second": 32.56,
"step": 16
},
{
"epoch": 5.0,
"learning_rate": 2.7777777777777783e-06,
"loss": 0.6562,
"step": 20
},
{
"epoch": 5.0,
"eval_accuracy": 0.875,
"eval_loss": 0.6495161056518555,
"eval_runtime": 0.1279,
"eval_samples_per_second": 125.074,
"eval_steps_per_second": 31.269,
"step": 20
},
{
"epoch": 6.0,
"eval_accuracy": 0.875,
"eval_loss": 0.667277455329895,
"eval_runtime": 0.1224,
"eval_samples_per_second": 130.678,
"eval_steps_per_second": 32.67,
"step": 24
},
{
"epoch": 7.0,
"eval_accuracy": 0.875,
"eval_loss": 0.6539220809936523,
"eval_runtime": 0.1364,
"eval_samples_per_second": 117.345,
"eval_steps_per_second": 29.336,
"step": 28
},
{
"epoch": 7.5,
"learning_rate": 1.3888888888888892e-06,
"loss": 0.6527,
"step": 30
},
{
"epoch": 8.0,
"eval_accuracy": 0.875,
"eval_loss": 0.6519389152526855,
"eval_runtime": 0.1233,
"eval_samples_per_second": 129.77,
"eval_steps_per_second": 32.443,
"step": 32
},
{
"epoch": 9.0,
"eval_accuracy": 0.875,
"eval_loss": 0.6602637767791748,
"eval_runtime": 0.1675,
"eval_samples_per_second": 95.523,
"eval_steps_per_second": 23.881,
"step": 36
},
{
"epoch": 10.0,
"learning_rate": 0.0,
"loss": 0.6596,
"step": 40
},
{
"epoch": 10.0,
"eval_accuracy": 0.875,
"eval_loss": 0.6525369882583618,
"eval_runtime": 0.1937,
"eval_samples_per_second": 82.597,
"eval_steps_per_second": 20.649,
"step": 40
},
{
"epoch": 10.0,
"step": 40,
"total_flos": 1.33783041595392e+16,
"train_loss": 0.6580723881721496,
"train_runtime": 37.1608,
"train_samples_per_second": 16.953,
"train_steps_per_second": 1.076
}
],
"max_steps": 40,
"num_train_epochs": 10,
"total_flos": 1.33783041595392e+16,
"trial_name": null,
"trial_params": null
}