tofu_grad-diff_1e-05_8_600_1187 / trainer_state.json
advit's picture
Upload folder using huggingface_hub
b90a0c8 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 150.0,
"eval_steps": 0,
"global_step": 150,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 12.0,
"grad_norm": 0.23003472871315364,
"learning_rate": 9.558232931726908e-06,
"loss": 0.0086,
"step": 12
},
{
"epoch": 24.0,
"grad_norm": 0.8898904454416547,
"learning_rate": 9.076305220883535e-06,
"loss": -0.0003,
"step": 24
},
{
"epoch": 36.0,
"grad_norm": 8.640425843085218,
"learning_rate": 8.594377510040161e-06,
"loss": -0.0696,
"step": 36
},
{
"epoch": 48.0,
"grad_norm": 5.337949748560155,
"learning_rate": 8.112449799196788e-06,
"loss": -0.2987,
"step": 48
},
{
"epoch": 60.0,
"grad_norm": 5.559062273744209,
"learning_rate": 7.630522088353415e-06,
"loss": -0.4952,
"step": 60
},
{
"epoch": 72.0,
"grad_norm": 6.350634142853266,
"learning_rate": 7.148594377510041e-06,
"loss": -0.7174,
"step": 72
},
{
"epoch": 84.0,
"grad_norm": 7.180350668235379,
"learning_rate": 6.666666666666667e-06,
"loss": -0.985,
"step": 84
},
{
"epoch": 96.0,
"grad_norm": 7.509864420415409,
"learning_rate": 6.184738955823294e-06,
"loss": -1.2969,
"step": 96
},
{
"epoch": 108.0,
"grad_norm": 7.5444188317099785,
"learning_rate": 5.70281124497992e-06,
"loss": -1.6125,
"step": 108
},
{
"epoch": 120.0,
"grad_norm": 7.179717586682728,
"learning_rate": 5.220883534136547e-06,
"loss": -1.9098,
"step": 120
},
{
"epoch": 132.0,
"grad_norm": 10.554415886756734,
"learning_rate": 4.7389558232931736e-06,
"loss": -2.2001,
"step": 132
},
{
"epoch": 144.0,
"grad_norm": 11.46525680143656,
"learning_rate": 4.2570281124497995e-06,
"loss": -2.5481,
"step": 144
}
],
"logging_steps": 12,
"max_steps": 250,
"num_input_tokens_seen": 0,
"num_train_epochs": 250,
"save_steps": 25,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}