{ "best_metric": 1.3690176010131836, "best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l59-l/checkpoint-11500", "epoch": 2.1186440677966103, "eval_steps": 500, "global_step": 11500, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.09211495946941783, "grad_norm": 20.952665328979492, "learning_rate": 4.884856300663228e-09, "loss": 0.3952, "step": 500 }, { "epoch": 0.09211495946941783, "eval_loss": 1.4940427541732788, "eval_runtime": 73.9282, "eval_samples_per_second": 16.327, "eval_steps_per_second": 2.043, "step": 500 }, { "epoch": 0.18422991893883567, "grad_norm": 59.28666687011719, "learning_rate": 4.769712601326456e-09, "loss": 0.4562, "step": 1000 }, { "epoch": 0.18422991893883567, "eval_loss": 1.4853259325027466, "eval_runtime": 74.6533, "eval_samples_per_second": 16.168, "eval_steps_per_second": 2.023, "step": 1000 }, { "epoch": 0.2763448784082535, "grad_norm": 1.179814338684082, "learning_rate": 4.654568901989683e-09, "loss": 0.5131, "step": 1500 }, { "epoch": 0.2763448784082535, "eval_loss": 1.4757832288742065, "eval_runtime": 76.2866, "eval_samples_per_second": 15.822, "eval_steps_per_second": 1.979, "step": 1500 }, { "epoch": 0.36845983787767134, "grad_norm": 0.900991678237915, "learning_rate": 4.5394252026529105e-09, "loss": 0.4481, "step": 2000 }, { "epoch": 0.36845983787767134, "eval_loss": 1.4675536155700684, "eval_runtime": 75.9348, "eval_samples_per_second": 15.895, "eval_steps_per_second": 1.989, "step": 2000 }, { "epoch": 0.46057479734708917, "grad_norm": 346.93853759765625, "learning_rate": 4.424281503316139e-09, "loss": 0.4839, "step": 2500 }, { "epoch": 0.46057479734708917, "eval_loss": 1.4585391283035278, "eval_runtime": 76.7708, "eval_samples_per_second": 15.722, "eval_steps_per_second": 1.967, "step": 2500 }, { "epoch": 0.552689756816507, "grad_norm": 318.47515869140625, "learning_rate": 4.3091378039793666e-09, "loss": 0.4377, "step": 3000 }, { "epoch": 0.552689756816507, "eval_loss": 1.4507631063461304, "eval_runtime": 76.5424, "eval_samples_per_second": 15.769, "eval_steps_per_second": 1.973, "step": 3000 }, { "epoch": 0.6448047162859248, "grad_norm": 112.41814422607422, "learning_rate": 4.193994104642594e-09, "loss": 0.4231, "step": 3500 }, { "epoch": 0.6448047162859248, "eval_loss": 1.4432125091552734, "eval_runtime": 76.4812, "eval_samples_per_second": 15.782, "eval_steps_per_second": 1.974, "step": 3500 }, { "epoch": 0.7369196757553427, "grad_norm": 83.51781463623047, "learning_rate": 4.078850405305822e-09, "loss": 0.4369, "step": 4000 }, { "epoch": 0.7369196757553427, "eval_loss": 1.4365661144256592, "eval_runtime": 76.4405, "eval_samples_per_second": 15.79, "eval_steps_per_second": 1.975, "step": 4000 }, { "epoch": 0.8290346352247605, "grad_norm": 460.5648498535156, "learning_rate": 3.963706705969049e-09, "loss": 0.4082, "step": 4500 }, { "epoch": 0.8290346352247605, "eval_loss": 1.4302303791046143, "eval_runtime": 75.042, "eval_samples_per_second": 16.084, "eval_steps_per_second": 2.012, "step": 4500 }, { "epoch": 0.9211495946941783, "grad_norm": 5.289137363433838, "learning_rate": 3.848563006632278e-09, "loss": 0.4234, "step": 5000 }, { "epoch": 0.9211495946941783, "eval_loss": 1.4243344068527222, "eval_runtime": 76.6198, "eval_samples_per_second": 15.753, "eval_steps_per_second": 1.971, "step": 5000 }, { "epoch": 1.013264554163596, "grad_norm": 0.6147650480270386, "learning_rate": 3.733419307295505e-09, "loss": 0.4266, "step": 5500 }, { "epoch": 1.013264554163596, "eval_loss": 1.4190788269042969, "eval_runtime": 75.4536, "eval_samples_per_second": 15.997, "eval_steps_per_second": 2.001, "step": 5500 }, { "epoch": 1.105379513633014, "grad_norm": 231.71815490722656, "learning_rate": 3.6182756079587326e-09, "loss": 0.4438, "step": 6000 }, { "epoch": 1.105379513633014, "eval_loss": 1.4136912822723389, "eval_runtime": 76.5978, "eval_samples_per_second": 15.758, "eval_steps_per_second": 1.971, "step": 6000 }, { "epoch": 1.1974944731024317, "grad_norm": 464.0763854980469, "learning_rate": 3.5031319086219602e-09, "loss": 0.3814, "step": 6500 }, { "epoch": 1.1974944731024317, "eval_loss": 1.4085203409194946, "eval_runtime": 75.4916, "eval_samples_per_second": 15.989, "eval_steps_per_second": 2.0, "step": 6500 }, { "epoch": 1.2896094325718497, "grad_norm": 2.5053436756134033, "learning_rate": 3.3879882092851882e-09, "loss": 0.3327, "step": 7000 }, { "epoch": 1.2896094325718497, "eval_loss": 1.4041872024536133, "eval_runtime": 75.2076, "eval_samples_per_second": 16.049, "eval_steps_per_second": 2.008, "step": 7000 }, { "epoch": 1.3817243920412676, "grad_norm": 126.44642639160156, "learning_rate": 3.2728445099484154e-09, "loss": 0.4045, "step": 7500 }, { "epoch": 1.3817243920412676, "eval_loss": 1.398864984512329, "eval_runtime": 75.3652, "eval_samples_per_second": 16.015, "eval_steps_per_second": 2.004, "step": 7500 }, { "epoch": 1.4738393515106853, "grad_norm": 1.046550989151001, "learning_rate": 3.1577008106116435e-09, "loss": 0.4038, "step": 8000 }, { "epoch": 1.4738393515106853, "eval_loss": 1.3937218189239502, "eval_runtime": 75.1378, "eval_samples_per_second": 16.064, "eval_steps_per_second": 2.01, "step": 8000 }, { "epoch": 1.565954310980103, "grad_norm": 212.23594665527344, "learning_rate": 3.042557111274871e-09, "loss": 0.3659, "step": 8500 }, { "epoch": 1.565954310980103, "eval_loss": 1.3894429206848145, "eval_runtime": 75.2403, "eval_samples_per_second": 16.042, "eval_steps_per_second": 2.007, "step": 8500 }, { "epoch": 1.658069270449521, "grad_norm": 2.719074010848999, "learning_rate": 2.927413411938099e-09, "loss": 0.4282, "step": 9000 }, { "epoch": 1.658069270449521, "eval_loss": 1.3854893445968628, "eval_runtime": 74.8392, "eval_samples_per_second": 16.128, "eval_steps_per_second": 2.018, "step": 9000 }, { "epoch": 1.750184229918939, "grad_norm": 4.810174465179443, "learning_rate": 2.8122697126013263e-09, "loss": 0.4173, "step": 9500 }, { "epoch": 1.750184229918939, "eval_loss": 1.3816269636154175, "eval_runtime": 75.6136, "eval_samples_per_second": 15.963, "eval_steps_per_second": 1.997, "step": 9500 }, { "epoch": 1.8422991893883567, "grad_norm": 630.7271118164062, "learning_rate": 2.6971260132645543e-09, "loss": 0.3758, "step": 10000 }, { "epoch": 1.8422991893883567, "eval_loss": 1.3779022693634033, "eval_runtime": 76.6948, "eval_samples_per_second": 15.738, "eval_steps_per_second": 1.969, "step": 10000 }, { "epoch": 1.9344141488577744, "grad_norm": 3.372347831726074, "learning_rate": 2.581982313927782e-09, "loss": 0.4105, "step": 10500 }, { "epoch": 1.9344141488577744, "eval_loss": 1.3744733333587646, "eval_runtime": 76.3833, "eval_samples_per_second": 15.802, "eval_steps_per_second": 1.977, "step": 10500 }, { "epoch": 2.026529108327192, "grad_norm": 0.4494275450706482, "learning_rate": 2.4668386145910095e-09, "loss": 0.3765, "step": 11000 }, { "epoch": 2.026529108327192, "eval_loss": 1.3716294765472412, "eval_runtime": 76.2721, "eval_samples_per_second": 15.825, "eval_steps_per_second": 1.98, "step": 11000 }, { "epoch": 2.1186440677966103, "grad_norm": 397.7524719238281, "learning_rate": 2.3516949152542375e-09, "loss": 0.3746, "step": 11500 }, { "epoch": 2.1186440677966103, "eval_loss": 1.3690176010131836, "eval_runtime": 74.7759, "eval_samples_per_second": 16.142, "eval_steps_per_second": 2.019, "step": 11500 } ], "logging_steps": 500, "max_steps": 21712, "num_input_tokens_seen": 0, "num_train_epochs": 4, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 4137940556192520.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }