{ "best_metric": 0.3466796875, "best_model_checkpoint": "/data/leeye/checkpoints_fft_t5_large/adversarial_qa_dbert_based_on/checkpoint-315", "epoch": 5.0, "eval_steps": 500, "global_step": 315, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.31746031746031744, "grad_norm": 18.570829391479492, "learning_rate": 0.00018730158730158731, "loss": 1.2416, "step": 20 }, { "epoch": 0.6349206349206349, "grad_norm": 15.798161506652832, "learning_rate": 0.00017460317460317462, "loss": 1.0892, "step": 40 }, { "epoch": 0.9523809523809523, "grad_norm": 7.127645015716553, "learning_rate": 0.00016190476190476192, "loss": 1.0162, "step": 60 }, { "epoch": 1.0, "eval_bleu": 0.2748783583138883, "eval_exact_match": 0.275390625, "eval_loss": 0.7607467174530029, "eval_runtime": 22.1309, "eval_samples_per_second": 45.186, "eval_steps_per_second": 0.723, "step": 63 }, { "epoch": 1.2698412698412698, "grad_norm": 6.7987775802612305, "learning_rate": 0.00014920634920634923, "loss": 0.4064, "step": 80 }, { "epoch": 1.5873015873015874, "grad_norm": 10.752936363220215, "learning_rate": 0.0001365079365079365, "loss": 0.3513, "step": 100 }, { "epoch": 1.9047619047619047, "grad_norm": 7.6475348472595215, "learning_rate": 0.0001238095238095238, "loss": 0.3929, "step": 120 }, { "epoch": 2.0, "eval_bleu": 0.24117803737423354, "eval_exact_match": 0.2958984375, "eval_loss": 0.794335126876831, "eval_runtime": 24.0083, "eval_samples_per_second": 41.652, "eval_steps_per_second": 0.666, "step": 126 }, { "epoch": 2.2222222222222223, "grad_norm": 2.862144947052002, "learning_rate": 0.00011111111111111112, "loss": 0.2161, "step": 140 }, { "epoch": 2.5396825396825395, "grad_norm": 5.910730838775635, "learning_rate": 9.841269841269841e-05, "loss": 0.138, "step": 160 }, { "epoch": 2.857142857142857, "grad_norm": 1.7475248575210571, "learning_rate": 8.571428571428571e-05, "loss": 0.1542, "step": 180 }, { "epoch": 3.0, "eval_bleu": 0.27200128946842234, "eval_exact_match": 0.3017578125, "eval_loss": 1.0052680969238281, "eval_runtime": 21.8202, "eval_samples_per_second": 45.829, "eval_steps_per_second": 0.733, "step": 189 }, { "epoch": 3.1746031746031744, "grad_norm": 2.242680549621582, "learning_rate": 7.301587301587302e-05, "loss": 0.095, "step": 200 }, { "epoch": 3.492063492063492, "grad_norm": 1.32904851436615, "learning_rate": 6.0317460317460316e-05, "loss": 0.0614, "step": 220 }, { "epoch": 3.8095238095238093, "grad_norm": 37.23604202270508, "learning_rate": 4.761904761904762e-05, "loss": 0.0544, "step": 240 }, { "epoch": 4.0, "eval_bleu": 0.31847923264557687, "eval_exact_match": 0.345703125, "eval_loss": 1.1005158424377441, "eval_runtime": 18.54, "eval_samples_per_second": 53.937, "eval_steps_per_second": 0.863, "step": 252 }, { "epoch": 4.1269841269841265, "grad_norm": 0.15852811932563782, "learning_rate": 3.492063492063492e-05, "loss": 0.0319, "step": 260 }, { "epoch": 4.444444444444445, "grad_norm": 0.4201483130455017, "learning_rate": 2.2222222222222223e-05, "loss": 0.0249, "step": 280 }, { "epoch": 4.761904761904762, "grad_norm": 0.46498769521713257, "learning_rate": 9.523809523809523e-06, "loss": 0.0239, "step": 300 }, { "epoch": 5.0, "eval_bleu": 0.30828732530539343, "eval_exact_match": 0.3466796875, "eval_loss": 1.138079285621643, "eval_runtime": 20.547, "eval_samples_per_second": 48.669, "eval_steps_per_second": 0.779, "step": 315 }, { "epoch": 5.0, "step": 315, "total_flos": 2586783250382848.0, "train_loss": 0.3393354445222824, "train_runtime": 835.7154, "train_samples_per_second": 2.991, "train_steps_per_second": 0.377 } ], "logging_steps": 20, "max_steps": 315, "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": true }, "attributes": {} } }, "total_flos": 2586783250382848.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }