VideoMAE_4_CLASS_QUALITY_CHECK / trainer_state.json
Shawon16's picture
End of training
d95bd39 verified
{
"best_metric": 1.0,
"best_model_checkpoint": "/media/cse/HDD/Shawon/shawon/10 fold timesformer/VideoMAE_4_CLASS_QUALITY_CHECK/checkpoint-30",
"epoch": 8.02,
"eval_steps": 500,
"global_step": 67,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.02,
"grad_norm": 17.483346939086914,
"learning_rate": 1e-05,
"loss": 6.0723,
"step": 7
},
{
"epoch": 0.02,
"eval_accuracy": 0.42857142857142855,
"eval_loss": 1.3153599500656128,
"eval_runtime": 2.0006,
"eval_samples_per_second": 6.998,
"eval_steps_per_second": 3.499,
"step": 7
},
{
"epoch": 1.0214285714285714,
"grad_norm": 20.505908966064453,
"learning_rate": 2.1428571428571428e-05,
"loss": 4.5479,
"step": 15
},
{
"epoch": 1.0214285714285714,
"eval_accuracy": 0.8571428571428571,
"eval_loss": 1.0319474935531616,
"eval_runtime": 1.9993,
"eval_samples_per_second": 7.003,
"eval_steps_per_second": 3.501,
"step": 15
},
{
"epoch": 2.02,
"grad_norm": 27.64762306213379,
"learning_rate": 3.142857142857143e-05,
"loss": 3.6957,
"step": 22
},
{
"epoch": 2.02,
"eval_accuracy": 0.8571428571428571,
"eval_loss": 0.6453508734703064,
"eval_runtime": 1.9837,
"eval_samples_per_second": 7.057,
"eval_steps_per_second": 3.529,
"step": 22
},
{
"epoch": 3.0214285714285714,
"grad_norm": 23.30771827697754,
"learning_rate": 4.1428571428571437e-05,
"loss": 1.7212,
"step": 30
},
{
"epoch": 3.0214285714285714,
"eval_accuracy": 1.0,
"eval_loss": 0.2796107828617096,
"eval_runtime": 1.9824,
"eval_samples_per_second": 7.062,
"eval_steps_per_second": 3.531,
"step": 30
},
{
"epoch": 4.02,
"grad_norm": 4.675124645233154,
"learning_rate": 4.9841269841269845e-05,
"loss": 0.6321,
"step": 37
},
{
"epoch": 4.02,
"eval_accuracy": 1.0,
"eval_loss": 0.05873652920126915,
"eval_runtime": 1.984,
"eval_samples_per_second": 7.056,
"eval_steps_per_second": 3.528,
"step": 37
},
{
"epoch": 5.021428571428571,
"grad_norm": 0.7413554787635803,
"learning_rate": 4.8571428571428576e-05,
"loss": 0.0861,
"step": 45
},
{
"epoch": 5.021428571428571,
"eval_accuracy": 1.0,
"eval_loss": 0.012020383961498737,
"eval_runtime": 1.9597,
"eval_samples_per_second": 7.144,
"eval_steps_per_second": 3.572,
"step": 45
},
{
"epoch": 6.02,
"grad_norm": 0.23269684612751007,
"learning_rate": 4.746031746031746e-05,
"loss": 0.0166,
"step": 52
},
{
"epoch": 6.02,
"eval_accuracy": 1.0,
"eval_loss": 0.002919810125604272,
"eval_runtime": 1.9438,
"eval_samples_per_second": 7.202,
"eval_steps_per_second": 3.601,
"step": 52
},
{
"epoch": 7.021428571428571,
"grad_norm": 0.06729763746261597,
"learning_rate": 4.6190476190476194e-05,
"loss": 0.0047,
"step": 60
},
{
"epoch": 7.021428571428571,
"eval_accuracy": 1.0,
"eval_loss": 0.0018126964569091797,
"eval_runtime": 2.0381,
"eval_samples_per_second": 6.869,
"eval_steps_per_second": 3.435,
"step": 60
},
{
"epoch": 8.02,
"grad_norm": 0.040539324283599854,
"learning_rate": 4.507936507936508e-05,
"loss": 0.0029,
"step": 67
},
{
"epoch": 8.02,
"eval_accuracy": 1.0,
"eval_loss": 0.0014999593840911984,
"eval_runtime": 1.9648,
"eval_samples_per_second": 7.125,
"eval_steps_per_second": 3.563,
"step": 67
},
{
"epoch": 8.02,
"step": 67,
"total_flos": 6.728868799001395e+17,
"train_loss": 1.8480051827074877,
"train_runtime": 179.1547,
"train_samples_per_second": 15.629,
"train_steps_per_second": 1.954
}
],
"logging_steps": 500,
"max_steps": 350,
"num_input_tokens_seen": 0,
"num_train_epochs": 9223372036854775807,
"save_steps": 500,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 5,
"early_stopping_threshold": 0.0
},
"attributes": {
"early_stopping_patience_counter": 5
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 6.728868799001395e+17,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}