{ "best_global_step": 200, "best_metric": 0.6800833940505981, "best_model_checkpoint": "./python-code-completion-ministral-8B/checkpoint-200", "epoch": 0.9411764705882353, "eval_steps": 50, "global_step": 200, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "entropy": 1.4469833582639695, "epoch": 0.047058823529411764, "grad_norm": 0.49369218945503235, "learning_rate": 9.999578688879085e-06, "loss": 1.7076, "mean_token_accuracy": 0.6342830918729305, "num_tokens": 25138.0, "step": 10 }, { "entropy": 1.430517853796482, "epoch": 0.09411764705882353, "grad_norm": 0.51407390832901, "learning_rate": 9.949107209404664e-06, "loss": 1.5636, "mean_token_accuracy": 0.6443205893039703, "num_tokens": 47578.0, "step": 20 }, { "entropy": 1.4371182411909103, "epoch": 0.1411764705882353, "grad_norm": 0.6639813184738159, "learning_rate": 9.815347070757234e-06, "loss": 1.4756, "mean_token_accuracy": 0.6693132996559144, "num_tokens": 71393.0, "step": 30 }, { "entropy": 1.3472976341843605, "epoch": 0.18823529411764706, "grad_norm": 0.3675422668457031, "learning_rate": 9.60054932593026e-06, "loss": 1.2256, "mean_token_accuracy": 0.6993286892771721, "num_tokens": 96359.0, "step": 40 }, { "entropy": 1.1503819346427917, "epoch": 0.23529411764705882, "grad_norm": 0.4163525104522705, "learning_rate": 9.30832881198487e-06, "loss": 1.0136, "mean_token_accuracy": 0.7729795873165131, "num_tokens": 123155.0, "step": 50 }, { "epoch": 0.23529411764705882, "eval_entropy": 1.0332347961266835, "eval_loss": 0.9837512373924255, "eval_mean_token_accuracy": 0.8083523658911387, "eval_num_tokens": 123155.0, "eval_runtime": 25.0181, "eval_samples_per_second": 5.996, "eval_steps_per_second": 5.996, "step": 50 }, { "entropy": 0.8932387962937355, "epoch": 0.2823529411764706, "grad_norm": 0.2984226942062378, "learning_rate": 8.943603315859101e-06, "loss": 0.8519, "mean_token_accuracy": 0.8314538225531578, "num_tokens": 149237.0, "step": 60 }, { "entropy": 0.7480494946241378, "epoch": 0.32941176470588235, "grad_norm": 0.409252405166626, "learning_rate": 8.512510812798426e-06, "loss": 0.7625, "mean_token_accuracy": 0.853736138343811, "num_tokens": 173954.0, "step": 70 }, { "entropy": 0.733599005267024, "epoch": 0.3764705882352941, "grad_norm": 0.3177991211414337, "learning_rate": 8.022306170204233e-06, "loss": 0.7384, "mean_token_accuracy": 0.8508381456136703, "num_tokens": 202672.0, "step": 80 }, { "entropy": 0.6522206105291843, "epoch": 0.4235294117647059, "grad_norm": 0.5717107057571411, "learning_rate": 7.481239055273959e-06, "loss": 0.6701, "mean_token_accuracy": 0.8760824427008629, "num_tokens": 224695.0, "step": 90 }, { "entropy": 0.7285707335919142, "epoch": 0.47058823529411764, "grad_norm": 0.21871985495090485, "learning_rate": 6.898415101128571e-06, "loss": 0.7319, "mean_token_accuracy": 0.8523965805768967, "num_tokens": 251370.0, "step": 100 }, { "epoch": 0.47058823529411764, "eval_entropy": 0.73099374294281, "eval_loss": 0.7096962332725525, "eval_mean_token_accuracy": 0.8550362022717793, "eval_num_tokens": 251370.0, "eval_runtime": 25.1112, "eval_samples_per_second": 5.973, "eval_steps_per_second": 5.973, "step": 100 }, { "entropy": 0.6658912554383278, "epoch": 0.5176470588235295, "grad_norm": 0.2736637592315674, "learning_rate": 6.283642667866317e-06, "loss": 0.6646, "mean_token_accuracy": 0.8663464844226837, "num_tokens": 276477.0, "step": 110 }, { "entropy": 0.6516228564083576, "epoch": 0.5647058823529412, "grad_norm": 0.2847215533256531, "learning_rate": 5.647267777405178e-06, "loss": 0.6503, "mean_token_accuracy": 0.8681117251515389, "num_tokens": 302513.0, "step": 120 }, { "entropy": 0.7009131267666817, "epoch": 0.611764705882353, "grad_norm": 0.22739574313163757, "learning_rate": 5e-06, "loss": 0.6881, "mean_token_accuracy": 0.8624396547675133, "num_tokens": 329679.0, "step": 130 }, { "entropy": 0.6468490872532129, "epoch": 0.6588235294117647, "grad_norm": 0.3506576716899872, "learning_rate": 4.352732222594823e-06, "loss": 0.6462, "mean_token_accuracy": 0.871060524880886, "num_tokens": 353837.0, "step": 140 }, { "entropy": 0.6587031967937946, "epoch": 0.7058823529411765, "grad_norm": 0.2826354205608368, "learning_rate": 3.7163573321336867e-06, "loss": 0.6779, "mean_token_accuracy": 0.8682561233639717, "num_tokens": 376786.0, "step": 150 }, { "epoch": 0.7058823529411765, "eval_entropy": 0.7037040118376414, "eval_loss": 0.6883785724639893, "eval_mean_token_accuracy": 0.8590577419598897, "eval_num_tokens": 376786.0, "eval_runtime": 25.1113, "eval_samples_per_second": 5.973, "eval_steps_per_second": 5.973, "step": 150 }, { "entropy": 0.6935035854578018, "epoch": 0.7529411764705882, "grad_norm": 0.32049205899238586, "learning_rate": 3.101584898871431e-06, "loss": 0.7223, "mean_token_accuracy": 0.8527676403522492, "num_tokens": 404164.0, "step": 160 }, { "entropy": 0.6355765651911497, "epoch": 0.8, "grad_norm": 0.27511146664619446, "learning_rate": 2.518760944726042e-06, "loss": 0.641, "mean_token_accuracy": 0.8753799632191658, "num_tokens": 427356.0, "step": 170 }, { "entropy": 0.6331917632371187, "epoch": 0.8470588235294118, "grad_norm": 0.289739191532135, "learning_rate": 1.977693829795769e-06, "loss": 0.646, "mean_token_accuracy": 0.8754393547773361, "num_tokens": 450321.0, "step": 180 }, { "entropy": 0.7034022852778434, "epoch": 0.8941176470588236, "grad_norm": 0.32424843311309814, "learning_rate": 1.4874891872015735e-06, "loss": 0.7084, "mean_token_accuracy": 0.857265616953373, "num_tokens": 478364.0, "step": 190 }, { "entropy": 0.6966812558472156, "epoch": 0.9411764705882353, "grad_norm": 0.27991771697998047, "learning_rate": 1.0563966841408995e-06, "loss": 0.7108, "mean_token_accuracy": 0.857306943833828, "num_tokens": 505339.0, "step": 200 }, { "epoch": 0.9411764705882353, "eval_entropy": 0.6970382163921992, "eval_loss": 0.6800833940505981, "eval_mean_token_accuracy": 0.8598487377166748, "eval_num_tokens": 505339.0, "eval_runtime": 25.145, "eval_samples_per_second": 5.965, "eval_steps_per_second": 5.965, "step": 200 } ], "logging_steps": 10, "max_steps": 250, "num_input_tokens_seen": 0, "num_train_epochs": 2, "save_steps": 100, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 2.321787690779443e+16, "train_batch_size": 1, "trial_name": null, "trial_params": null }