WilliamHH's picture
Training in progress, step 500
e9d35ba verified
{
"best_global_step": 500,
"best_metric": 0.6847298741340637,
"best_model_checkpoint": "Assignment4_Distilled_ModernBERT/run-4/checkpoint-500",
"epoch": 1.0482180293501049,
"eval_steps": 100,
"global_step": 500,
"is_hyper_param_search": true,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.20964360587002095,
"grad_norm": 11.635396957397461,
"learning_rate": 5.974527645836366e-05,
"loss": 6.4293,
"step": 100
},
{
"epoch": 0.20964360587002095,
"eval_accuracy": 0.7254838709677419,
"eval_loss": 3.423161268234253,
"eval_runtime": 24.7311,
"eval_samples_per_second": 125.348,
"eval_steps_per_second": 15.689,
"step": 100
},
{
"epoch": 0.4192872117400419,
"grad_norm": 7.08547830581665,
"learning_rate": 5.897521628664326e-05,
"loss": 2.4649,
"step": 200
},
{
"epoch": 0.4192872117400419,
"eval_accuracy": 0.9012903225806451,
"eval_loss": 1.6392749547958374,
"eval_runtime": 24.334,
"eval_samples_per_second": 127.394,
"eval_steps_per_second": 15.945,
"step": 200
},
{
"epoch": 0.6289308176100629,
"grad_norm": 2.8314898014068604,
"learning_rate": 5.7703135417326854e-05,
"loss": 1.364,
"step": 300
},
{
"epoch": 0.6289308176100629,
"eval_accuracy": 0.9361290322580645,
"eval_loss": 1.0801855325698853,
"eval_runtime": 24.4718,
"eval_samples_per_second": 126.676,
"eval_steps_per_second": 15.855,
"step": 300
},
{
"epoch": 0.8385744234800838,
"grad_norm": 11.189908981323242,
"learning_rate": 5.595107375234632e-05,
"loss": 0.9328,
"step": 400
},
{
"epoch": 0.8385744234800838,
"eval_accuracy": 0.9506451612903226,
"eval_loss": 0.8573997616767883,
"eval_runtime": 24.7134,
"eval_samples_per_second": 125.438,
"eval_steps_per_second": 15.7,
"step": 400
},
{
"epoch": 1.0482180293501049,
"grad_norm": 9.433755874633789,
"learning_rate": 5.3749387276298395e-05,
"loss": 0.708,
"step": 500
},
{
"epoch": 1.0482180293501049,
"eval_accuracy": 0.9564516129032258,
"eval_loss": 0.6847298741340637,
"eval_runtime": 24.9978,
"eval_samples_per_second": 124.011,
"eval_steps_per_second": 15.521,
"step": 500
}
],
"logging_steps": 100,
"max_steps": 2385,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"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": 224151802852992.0,
"train_batch_size": 32,
"trial_name": null,
"trial_params": {
"alpha": 0.5611126477660928,
"num_train_epochs": 5,
"temperature": 7
}
}