Model-J__DINO__model_idx_0911 / trainer_state.json
Eliahu's picture
Upload folder using huggingface_hub
f5c56fa verified
{
"best_metric": 0.9258666666666666,
"best_model_checkpoint": "./vit_finetuned_models_dataset/CIFAR100/50_from_100/facebook_dino-vitb16/model_idx_0911/checkpoints/checkpoint-2997",
"epoch": 9.0,
"eval_steps": 500,
"global_step": 2997,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 50.27267837524414,
"learning_rate": 8.003003003003004e-05,
"loss": 0.9111,
"step": 333
},
{
"epoch": 1.0,
"eval_accuracy": 0.8354666666666667,
"eval_loss": 0.5687533617019653,
"eval_runtime": 8.7172,
"eval_samples_per_second": 430.182,
"eval_steps_per_second": 6.768,
"step": 333
},
{
"epoch": 2.0,
"grad_norm": 24.453411102294922,
"learning_rate": 7.003003003003004e-05,
"loss": 0.3033,
"step": 666
},
{
"epoch": 2.0,
"eval_accuracy": 0.872,
"eval_loss": 0.42214518785476685,
"eval_runtime": 8.367,
"eval_samples_per_second": 448.189,
"eval_steps_per_second": 7.052,
"step": 666
},
{
"epoch": 3.0,
"grad_norm": 0.04773509129881859,
"learning_rate": 6.003003003003003e-05,
"loss": 0.1659,
"step": 999
},
{
"epoch": 3.0,
"eval_accuracy": 0.8922666666666667,
"eval_loss": 0.39090031385421753,
"eval_runtime": 7.4563,
"eval_samples_per_second": 502.932,
"eval_steps_per_second": 7.913,
"step": 999
},
{
"epoch": 4.0,
"grad_norm": 11.433270454406738,
"learning_rate": 5.003003003003004e-05,
"loss": 0.0942,
"step": 1332
},
{
"epoch": 4.0,
"eval_accuracy": 0.8818666666666667,
"eval_loss": 0.4659087657928467,
"eval_runtime": 7.4588,
"eval_samples_per_second": 502.76,
"eval_steps_per_second": 7.91,
"step": 1332
},
{
"epoch": 5.0,
"grad_norm": 0.025731440633535385,
"learning_rate": 4.006006006006007e-05,
"loss": 0.0522,
"step": 1665
},
{
"epoch": 5.0,
"eval_accuracy": 0.9024,
"eval_loss": 0.40758001804351807,
"eval_runtime": 7.3789,
"eval_samples_per_second": 508.208,
"eval_steps_per_second": 7.996,
"step": 1665
},
{
"epoch": 6.0,
"grad_norm": 0.04738115519285202,
"learning_rate": 3.0060060060060065e-05,
"loss": 0.0262,
"step": 1998
},
{
"epoch": 6.0,
"eval_accuracy": 0.9082666666666667,
"eval_loss": 0.3946199119091034,
"eval_runtime": 7.4341,
"eval_samples_per_second": 504.434,
"eval_steps_per_second": 7.936,
"step": 1998
},
{
"epoch": 7.0,
"grad_norm": 0.0012796444352716208,
"learning_rate": 2.0060060060060062e-05,
"loss": 0.011,
"step": 2331
},
{
"epoch": 7.0,
"eval_accuracy": 0.9210666666666667,
"eval_loss": 0.35366639494895935,
"eval_runtime": 7.2374,
"eval_samples_per_second": 518.141,
"eval_steps_per_second": 8.152,
"step": 2331
},
{
"epoch": 8.0,
"grad_norm": 0.05798524618148804,
"learning_rate": 1.0060060060060061e-05,
"loss": 0.0023,
"step": 2664
},
{
"epoch": 8.0,
"eval_accuracy": 0.9205333333333333,
"eval_loss": 0.3412811756134033,
"eval_runtime": 7.3964,
"eval_samples_per_second": 507.002,
"eval_steps_per_second": 7.977,
"step": 2664
},
{
"epoch": 9.0,
"grad_norm": 0.0002840954693965614,
"learning_rate": 6.006006006006006e-08,
"loss": 0.0006,
"step": 2997
},
{
"epoch": 9.0,
"eval_accuracy": 0.9258666666666666,
"eval_loss": 0.3350273072719574,
"eval_runtime": 7.271,
"eval_samples_per_second": 515.748,
"eval_steps_per_second": 8.114,
"step": 2997
}
],
"logging_steps": 500,
"max_steps": 2997,
"num_input_tokens_seen": 0,
"num_train_epochs": 9,
"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": 1.482671885603328e+19,
"train_batch_size": 64,
"trial_name": null,
"trial_params": null
}