Model-J__DINO__model_idx_0678 / trainer_state.json
Eliahu's picture
Upload folder using huggingface_hub
7ad2963 verified
{
"best_metric": 0.9181333333333334,
"best_model_checkpoint": "./vit_finetuned_models_dataset/CIFAR100/50_from_100/facebook_dino-vitb16/model_idx_0678/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": 24.114116668701172,
"learning_rate": 8.006006006006006e-05,
"loss": 0.946,
"step": 333
},
{
"epoch": 1.0,
"eval_accuracy": 0.8352,
"eval_loss": 0.5771884322166443,
"eval_runtime": 12.6535,
"eval_samples_per_second": 296.36,
"eval_steps_per_second": 4.663,
"step": 333
},
{
"epoch": 2.0,
"grad_norm": 3.7141811847686768,
"learning_rate": 7.006006006006006e-05,
"loss": 0.321,
"step": 666
},
{
"epoch": 2.0,
"eval_accuracy": 0.8456,
"eval_loss": 0.5261477828025818,
"eval_runtime": 11.7723,
"eval_samples_per_second": 318.545,
"eval_steps_per_second": 5.012,
"step": 666
},
{
"epoch": 3.0,
"grad_norm": 0.22823549807071686,
"learning_rate": 6.006006006006006e-05,
"loss": 0.1726,
"step": 999
},
{
"epoch": 3.0,
"eval_accuracy": 0.8717333333333334,
"eval_loss": 0.47572076320648193,
"eval_runtime": 11.5478,
"eval_samples_per_second": 324.737,
"eval_steps_per_second": 5.109,
"step": 999
},
{
"epoch": 4.0,
"grad_norm": 160.7298126220703,
"learning_rate": 5.0060060060060066e-05,
"loss": 0.1034,
"step": 1332
},
{
"epoch": 4.0,
"eval_accuracy": 0.8741333333333333,
"eval_loss": 0.4934878349304199,
"eval_runtime": 11.9821,
"eval_samples_per_second": 312.966,
"eval_steps_per_second": 4.924,
"step": 1332
},
{
"epoch": 5.0,
"grad_norm": 0.08969806879758835,
"learning_rate": 4.006006006006007e-05,
"loss": 0.0573,
"step": 1665
},
{
"epoch": 5.0,
"eval_accuracy": 0.8797333333333334,
"eval_loss": 0.49537256360054016,
"eval_runtime": 11.5058,
"eval_samples_per_second": 325.924,
"eval_steps_per_second": 5.128,
"step": 1665
},
{
"epoch": 6.0,
"grad_norm": 36.12517547607422,
"learning_rate": 3.0060060060060065e-05,
"loss": 0.0315,
"step": 1998
},
{
"epoch": 6.0,
"eval_accuracy": 0.8965333333333333,
"eval_loss": 0.46518462896347046,
"eval_runtime": 11.7734,
"eval_samples_per_second": 318.514,
"eval_steps_per_second": 5.011,
"step": 1998
},
{
"epoch": 7.0,
"grad_norm": 0.0001624671567697078,
"learning_rate": 2.0060060060060062e-05,
"loss": 0.0124,
"step": 2331
},
{
"epoch": 7.0,
"eval_accuracy": 0.9050666666666667,
"eval_loss": 0.44985175132751465,
"eval_runtime": 12.109,
"eval_samples_per_second": 309.688,
"eval_steps_per_second": 4.872,
"step": 2331
},
{
"epoch": 8.0,
"grad_norm": 0.007620618212968111,
"learning_rate": 1.0060060060060061e-05,
"loss": 0.0045,
"step": 2664
},
{
"epoch": 8.0,
"eval_accuracy": 0.9136,
"eval_loss": 0.4013453423976898,
"eval_runtime": 11.6389,
"eval_samples_per_second": 322.196,
"eval_steps_per_second": 5.069,
"step": 2664
},
{
"epoch": 9.0,
"grad_norm": 8.350544521817937e-05,
"learning_rate": 6.006006006006006e-08,
"loss": 0.001,
"step": 2997
},
{
"epoch": 9.0,
"eval_accuracy": 0.9181333333333334,
"eval_loss": 0.3817403316497803,
"eval_runtime": 11.6139,
"eval_samples_per_second": 322.888,
"eval_steps_per_second": 5.08,
"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
}