gemma4-e4b-webvid4K_FT / trainer_state.json
bear7011's picture
Upload full fine-tuned Gemma model and README
463a947 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 500,
"global_step": 124,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.08113590263691683,
"grad_norm": 56.5714225769043,
"learning_rate": 4.978612153434527e-06,
"loss": 4.808269882202149,
"step": 10
},
{
"epoch": 0.16227180527383367,
"grad_norm": 20.036739349365234,
"learning_rate": 4.809698831278217e-06,
"loss": 1.8294570922851563,
"step": 20
},
{
"epoch": 0.2434077079107505,
"grad_norm": 12.78960132598877,
"learning_rate": 4.4833833507280884e-06,
"loss": 1.760553741455078,
"step": 30
},
{
"epoch": 0.32454361054766734,
"grad_norm": 10.83691692352295,
"learning_rate": 4.021903572521802e-06,
"loss": 1.756688690185547,
"step": 40
},
{
"epoch": 0.4056795131845842,
"grad_norm": 10.447757720947266,
"learning_rate": 3.4567085809127247e-06,
"loss": 1.73138427734375,
"step": 50
},
{
"epoch": 0.486815415821501,
"grad_norm": 12.645753860473633,
"learning_rate": 2.82631548055013e-06,
"loss": 1.6285388946533204,
"step": 60
},
{
"epoch": 0.5679513184584178,
"grad_norm": 12.400439262390137,
"learning_rate": 2.173684519449872e-06,
"loss": 1.6827095031738282,
"step": 70
},
{
"epoch": 0.6490872210953347,
"grad_norm": 15.744965553283691,
"learning_rate": 1.5432914190872757e-06,
"loss": 1.6250295639038086,
"step": 80
},
{
"epoch": 0.7302231237322515,
"grad_norm": 15.432747840881348,
"learning_rate": 9.780964274781984e-07,
"loss": 1.6873130798339844,
"step": 90
},
{
"epoch": 0.8113590263691683,
"grad_norm": 9.620952606201172,
"learning_rate": 5.166166492719124e-07,
"loss": 1.7171627044677735,
"step": 100
},
{
"epoch": 0.8924949290060852,
"grad_norm": 8.799121856689453,
"learning_rate": 1.9030116872178317e-07,
"loss": 1.6061542510986329,
"step": 110
},
{
"epoch": 0.973630831643002,
"grad_norm": 9.194286346435547,
"learning_rate": 2.1387846565474047e-08,
"loss": 1.6738105773925782,
"step": 120
},
{
"epoch": 1.0,
"step": 124,
"total_flos": 770301428236288.0,
"train_loss": 1.9510461630359772,
"train_runtime": 21182.0803,
"train_samples_per_second": 0.186,
"train_steps_per_second": 0.006
}
],
"logging_steps": 10,
"max_steps": 124,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"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": 770301428236288.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}