{ "best_metric": 0.9344, "best_model_checkpoint": "./vit_finetuned_models_dataset/CIFAR100/50_from_100/facebook_dino-vitb16/model_idx_0108/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": 1.9157744646072388, "learning_rate": 4.850126564054637e-05, "loss": 0.7427, "step": 333 }, { "epoch": 1.0, "eval_accuracy": 0.8786666666666667, "eval_loss": 0.39399003982543945, "eval_runtime": 4.0751, "eval_samples_per_second": 920.212, "eval_steps_per_second": 14.478, "step": 333 }, { "epoch": 2.0, "grad_norm": 0.027789583429694176, "learning_rate": 4.416794553834106e-05, "loss": 0.213, "step": 666 }, { "epoch": 2.0, "eval_accuracy": 0.8949333333333334, "eval_loss": 0.33196935057640076, "eval_runtime": 4.0549, "eval_samples_per_second": 924.806, "eval_steps_per_second": 14.55, "step": 666 }, { "epoch": 3.0, "grad_norm": Infinity, "learning_rate": 3.7545362870758595e-05, "loss": 0.1029, "step": 999 }, { "epoch": 3.0, "eval_accuracy": 0.8933333333333333, "eval_loss": 0.35182154178619385, "eval_runtime": 4.1966, "eval_samples_per_second": 893.586, "eval_steps_per_second": 14.059, "step": 999 }, { "epoch": 4.0, "grad_norm": 0.018823252990841866, "learning_rate": 2.9392810892864702e-05, "loss": 0.0443, "step": 1332 }, { "epoch": 4.0, "eval_accuracy": 0.9106666666666666, "eval_loss": 0.33442744612693787, "eval_runtime": 4.1719, "eval_samples_per_second": 898.88, "eval_steps_per_second": 14.142, "step": 1332 }, { "epoch": 5.0, "grad_norm": 0.2509842813014984, "learning_rate": 2.0710421090307246e-05, "loss": 0.0231, "step": 1665 }, { "epoch": 5.0, "eval_accuracy": 0.9213333333333333, "eval_loss": 0.31589528918266296, "eval_runtime": 4.0719, "eval_samples_per_second": 920.948, "eval_steps_per_second": 14.49, "step": 1665 }, { "epoch": 6.0, "grad_norm": 0.00012362049892544746, "learning_rate": 1.2545417811701016e-05, "loss": 0.0055, "step": 1998 }, { "epoch": 6.0, "eval_accuracy": 0.9296, "eval_loss": 0.3054322302341461, "eval_runtime": 4.4158, "eval_samples_per_second": 849.224, "eval_steps_per_second": 13.361, "step": 1998 }, { "epoch": 7.0, "grad_norm": 0.09116710722446442, "learning_rate": 5.882620955060414e-06, "loss": 0.0005, "step": 2331 }, { "epoch": 7.0, "eval_accuracy": 0.9341333333333334, "eval_loss": 0.2975241243839264, "eval_runtime": 4.0686, "eval_samples_per_second": 921.695, "eval_steps_per_second": 14.501, "step": 2331 }, { "epoch": 8.0, "grad_norm": 0.0016982945380732417, "learning_rate": 1.5256621537052152e-06, "loss": 0.0001, "step": 2664 }, { "epoch": 8.0, "eval_accuracy": 0.9330666666666667, "eval_loss": 0.295346736907959, "eval_runtime": 4.0577, "eval_samples_per_second": 924.174, "eval_steps_per_second": 14.54, "step": 2664 }, { "epoch": 9.0, "grad_norm": 0.00819256529211998, "learning_rate": 5.494094242330938e-11, "loss": 0.0001, "step": 2997 }, { "epoch": 9.0, "eval_accuracy": 0.9344, "eval_loss": 0.2916950583457947, "eval_runtime": 4.1225, "eval_samples_per_second": 909.648, "eval_steps_per_second": 14.312, "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 }