ds3-img-classification / checkpoint-64 /trainer_state.json
jnmrr's picture
Upload model with metrics - 2024-12-12 01:56
d3a7216 verified
{
"best_metric": 0.1350535273126544,
"best_model_checkpoint": "ds3-img-classification/checkpoint-35",
"epoch": 0.9922480620155039,
"eval_steps": 35,
"global_step": 64,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.015503875968992248,
"grad_norm": 16.251585006713867,
"learning_rate": 0.0002857142857142857,
"loss": 6.0665,
"step": 1
},
{
"epoch": 0.5426356589147286,
"grad_norm": 5.580934524536133,
"learning_rate": 0.0010275543423681622,
"loss": 5.2299,
"step": 35
},
{
"epoch": 0.5426356589147286,
"eval_accuracy": 0.1350535273126544,
"eval_loss": 5.006561279296875,
"eval_macro_f1": 0.038001266460012215,
"eval_macro_precision": 0.033336047367919855,
"eval_macro_recall": 0.06460349403258135,
"eval_micro_f1": 0.1350535273126544,
"eval_micro_precision": 0.1350535273126544,
"eval_micro_recall": 0.1350535273126544,
"eval_runtime": 8.2574,
"eval_samples_per_second": 441.182,
"eval_steps_per_second": 3.512,
"eval_weighted_f1": 0.0662231556079451,
"eval_weighted_precision": 0.054814565244902975,
"eval_weighted_recall": 0.1350535273126544,
"step": 35
}
],
"logging_steps": 35,
"max_steps": 64,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 35,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 8.978215898519175e+18,
"train_batch_size": 256,
"trial_name": null,
"trial_params": null
}