| { | |
| "best_metric": 0.70501708984375, | |
| "best_model_checkpoint": "vit-base-patch16-224-in21k-rotated-dungeons-v11/checkpoint-300", | |
| "epoch": 27.0, | |
| "eval_steps": 100, | |
| "global_step": 324, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.8333333333333334, | |
| "grad_norm": 0.7690829634666443, | |
| "learning_rate": 9.691358024691358e-06, | |
| "loss": 0.1644, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 1.6666666666666665, | |
| "grad_norm": 5.240622520446777, | |
| "learning_rate": 9.382716049382717e-06, | |
| "loss": 0.1883, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 2.5, | |
| "grad_norm": 1.536397099494934, | |
| "learning_rate": 9.074074074074075e-06, | |
| "loss": 0.1732, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 3.3333333333333335, | |
| "grad_norm": 1.8029364347457886, | |
| "learning_rate": 8.765432098765432e-06, | |
| "loss": 0.1666, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 4.166666666666667, | |
| "grad_norm": 0.6660683155059814, | |
| "learning_rate": 8.456790123456791e-06, | |
| "loss": 0.1536, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "grad_norm": 0.8398900032043457, | |
| "learning_rate": 8.148148148148148e-06, | |
| "loss": 0.1528, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 5.833333333333333, | |
| "grad_norm": 0.7446826696395874, | |
| "learning_rate": 7.839506172839507e-06, | |
| "loss": 0.1502, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 6.666666666666667, | |
| "grad_norm": 1.3342101573944092, | |
| "learning_rate": 7.530864197530865e-06, | |
| "loss": 0.15, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 7.5, | |
| "grad_norm": 1.0920078754425049, | |
| "learning_rate": 7.222222222222223e-06, | |
| "loss": 0.1494, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 8.333333333333334, | |
| "grad_norm": 0.5975252389907837, | |
| "learning_rate": 6.913580246913581e-06, | |
| "loss": 0.1396, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 8.333333333333334, | |
| "eval_accuracy": 0.875, | |
| "eval_loss": 0.7252655029296875, | |
| "eval_runtime": 0.095, | |
| "eval_samples_per_second": 84.236, | |
| "eval_steps_per_second": 10.529, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 9.166666666666666, | |
| "grad_norm": 0.576201856136322, | |
| "learning_rate": 6.60493827160494e-06, | |
| "loss": 0.1407, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "grad_norm": 0.5525928735733032, | |
| "learning_rate": 6.296296296296297e-06, | |
| "loss": 0.1399, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 10.833333333333334, | |
| "grad_norm": 0.8266479969024658, | |
| "learning_rate": 5.9876543209876546e-06, | |
| "loss": 0.1365, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 11.666666666666666, | |
| "grad_norm": 0.6235507130622864, | |
| "learning_rate": 5.6790123456790125e-06, | |
| "loss": 0.1299, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 12.5, | |
| "grad_norm": 0.5794048309326172, | |
| "learning_rate": 5.370370370370371e-06, | |
| "loss": 0.1318, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 13.333333333333334, | |
| "grad_norm": 0.4382638931274414, | |
| "learning_rate": 5.061728395061729e-06, | |
| "loss": 0.1328, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 14.166666666666666, | |
| "grad_norm": 0.5124505758285522, | |
| "learning_rate": 4.753086419753087e-06, | |
| "loss": 0.1201, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 15.0, | |
| "grad_norm": 0.4952060878276825, | |
| "learning_rate": 4.444444444444444e-06, | |
| "loss": 0.1289, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 15.833333333333334, | |
| "grad_norm": 0.5130446553230286, | |
| "learning_rate": 4.135802469135803e-06, | |
| "loss": 0.126, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 16.666666666666668, | |
| "grad_norm": 0.47538742423057556, | |
| "learning_rate": 3.827160493827161e-06, | |
| "loss": 0.1215, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 16.666666666666668, | |
| "eval_accuracy": 0.875, | |
| "eval_loss": 0.71337890625, | |
| "eval_runtime": 0.102, | |
| "eval_samples_per_second": 78.404, | |
| "eval_steps_per_second": 9.801, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 17.5, | |
| "grad_norm": 0.4837852120399475, | |
| "learning_rate": 3.5185185185185187e-06, | |
| "loss": 0.1286, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 18.333333333333332, | |
| "grad_norm": 0.472771018743515, | |
| "learning_rate": 3.2098765432098767e-06, | |
| "loss": 0.1191, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 19.166666666666668, | |
| "grad_norm": 0.39906787872314453, | |
| "learning_rate": 2.901234567901235e-06, | |
| "loss": 0.1156, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "grad_norm": 0.54490727186203, | |
| "learning_rate": 2.5925925925925925e-06, | |
| "loss": 0.1241, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 20.833333333333332, | |
| "grad_norm": 0.5751241445541382, | |
| "learning_rate": 2.283950617283951e-06, | |
| "loss": 0.118, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 21.666666666666668, | |
| "grad_norm": 0.4648211598396301, | |
| "learning_rate": 1.9753086419753087e-06, | |
| "loss": 0.1257, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 22.5, | |
| "grad_norm": 0.46896305680274963, | |
| "learning_rate": 1.6666666666666667e-06, | |
| "loss": 0.1183, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 23.333333333333332, | |
| "grad_norm": 0.4199012517929077, | |
| "learning_rate": 1.3580246913580248e-06, | |
| "loss": 0.116, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 24.166666666666668, | |
| "grad_norm": 0.45512181520462036, | |
| "learning_rate": 1.0493827160493827e-06, | |
| "loss": 0.1164, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 25.0, | |
| "grad_norm": 0.527928352355957, | |
| "learning_rate": 7.407407407407407e-07, | |
| "loss": 0.1175, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 25.0, | |
| "eval_accuracy": 0.875, | |
| "eval_loss": 0.70501708984375, | |
| "eval_runtime": 0.0737, | |
| "eval_samples_per_second": 108.51, | |
| "eval_steps_per_second": 13.564, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 25.833333333333332, | |
| "grad_norm": 0.3864317834377289, | |
| "learning_rate": 4.320987654320988e-07, | |
| "loss": 0.1147, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 26.666666666666668, | |
| "grad_norm": 0.46536338329315186, | |
| "learning_rate": 1.234567901234568e-07, | |
| "loss": 0.1191, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 27.0, | |
| "step": 324, | |
| "total_flos": 9.834174165305549e+16, | |
| "train_loss": 0.13507753610610962, | |
| "train_runtime": 115.2462, | |
| "train_samples_per_second": 11.011, | |
| "train_steps_per_second": 2.811 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 324, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 27, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 9.834174165305549e+16, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |