| { |
| "best_metric": 0.8933333333333333, |
| "best_model_checkpoint": "policy_gte_large_5/checkpoint-3", |
| "epoch": 5.0, |
| "eval_steps": 500, |
| "global_step": 5, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 1.0, |
| "eval_cosine_accuracy@1": 0.8266666666666667, |
| "eval_cosine_accuracy@10": 1.0, |
| "eval_cosine_accuracy@3": 0.9866666666666667, |
| "eval_cosine_accuracy@5": 0.9866666666666667, |
| "eval_cosine_map@100": 0.9022222222222223, |
| "eval_cosine_mrr@10": 0.9022222222222223, |
| "eval_cosine_ndcg@10": 0.9271276662566819, |
| "eval_cosine_precision@1": 0.8266666666666667, |
| "eval_cosine_precision@10": 0.09999999999999998, |
| "eval_cosine_precision@3": 0.3288888888888888, |
| "eval_cosine_precision@5": 0.1973333333333333, |
| "eval_cosine_recall@1": 0.8266666666666667, |
| "eval_cosine_recall@10": 1.0, |
| "eval_cosine_recall@3": 0.9866666666666667, |
| "eval_cosine_recall@5": 0.9866666666666667, |
| "eval_dot_accuracy@1": 0.8266666666666667, |
| "eval_dot_accuracy@10": 1.0, |
| "eval_dot_accuracy@3": 0.9866666666666667, |
| "eval_dot_accuracy@5": 0.9866666666666667, |
| "eval_dot_map@100": 0.9019047619047619, |
| "eval_dot_mrr@10": 0.901904761904762, |
| "eval_dot_ndcg@10": 0.926822681539686, |
| "eval_dot_precision@1": 0.8266666666666667, |
| "eval_dot_precision@10": 0.09999999999999998, |
| "eval_dot_precision@3": 0.3288888888888888, |
| "eval_dot_precision@5": 0.1973333333333333, |
| "eval_dot_recall@1": 0.8266666666666667, |
| "eval_dot_recall@10": 1.0, |
| "eval_dot_recall@3": 0.9866666666666667, |
| "eval_dot_recall@5": 0.9866666666666667, |
| "eval_runtime": 0.2111, |
| "eval_samples_per_second": 0.0, |
| "eval_steps_per_second": 0.0, |
| "step": 1 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_cosine_accuracy@1": 0.88, |
| "eval_cosine_accuracy@10": 1.0, |
| "eval_cosine_accuracy@3": 0.9866666666666667, |
| "eval_cosine_accuracy@5": 0.9866666666666667, |
| "eval_cosine_map@100": 0.931111111111111, |
| "eval_cosine_mrr@10": 0.9311111111111112, |
| "eval_cosine_ndcg@10": 0.9485571427804901, |
| "eval_cosine_precision@1": 0.88, |
| "eval_cosine_precision@10": 0.09999999999999998, |
| "eval_cosine_precision@3": 0.3288888888888888, |
| "eval_cosine_precision@5": 0.1973333333333333, |
| "eval_cosine_recall@1": 0.88, |
| "eval_cosine_recall@10": 1.0, |
| "eval_cosine_recall@3": 0.9866666666666667, |
| "eval_cosine_recall@5": 0.9866666666666667, |
| "eval_dot_accuracy@1": 0.88, |
| "eval_dot_accuracy@10": 1.0, |
| "eval_dot_accuracy@3": 0.9866666666666667, |
| "eval_dot_accuracy@5": 0.9866666666666667, |
| "eval_dot_map@100": 0.931111111111111, |
| "eval_dot_mrr@10": 0.9311111111111112, |
| "eval_dot_ndcg@10": 0.9485571427804901, |
| "eval_dot_precision@1": 0.88, |
| "eval_dot_precision@10": 0.09999999999999998, |
| "eval_dot_precision@3": 0.3288888888888888, |
| "eval_dot_precision@5": 0.1973333333333333, |
| "eval_dot_recall@1": 0.88, |
| "eval_dot_recall@10": 1.0, |
| "eval_dot_recall@3": 0.9866666666666667, |
| "eval_dot_recall@5": 0.9866666666666667, |
| "eval_runtime": 0.2179, |
| "eval_samples_per_second": 0.0, |
| "eval_steps_per_second": 0.0, |
| "step": 2 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_cosine_accuracy@1": 0.8933333333333333, |
| "eval_cosine_accuracy@10": 1.0, |
| "eval_cosine_accuracy@3": 0.9866666666666667, |
| "eval_cosine_accuracy@5": 0.9866666666666667, |
| "eval_cosine_map@100": 0.9396825396825398, |
| "eval_cosine_mrr@10": 0.9396825396825398, |
| "eval_cosine_ndcg@10": 0.954918824730161, |
| "eval_cosine_precision@1": 0.8933333333333333, |
| "eval_cosine_precision@10": 0.09999999999999998, |
| "eval_cosine_precision@3": 0.3288888888888888, |
| "eval_cosine_precision@5": 0.1973333333333333, |
| "eval_cosine_recall@1": 0.8933333333333333, |
| "eval_cosine_recall@10": 1.0, |
| "eval_cosine_recall@3": 0.9866666666666667, |
| "eval_cosine_recall@5": 0.9866666666666667, |
| "eval_dot_accuracy@1": 0.8933333333333333, |
| "eval_dot_accuracy@10": 1.0, |
| "eval_dot_accuracy@3": 0.9866666666666667, |
| "eval_dot_accuracy@5": 0.9866666666666667, |
| "eval_dot_map@100": 0.9396825396825398, |
| "eval_dot_mrr@10": 0.9396825396825398, |
| "eval_dot_ndcg@10": 0.954918824730161, |
| "eval_dot_precision@1": 0.8933333333333333, |
| "eval_dot_precision@10": 0.09999999999999998, |
| "eval_dot_precision@3": 0.3288888888888888, |
| "eval_dot_precision@5": 0.1973333333333333, |
| "eval_dot_recall@1": 0.8933333333333333, |
| "eval_dot_recall@10": 1.0, |
| "eval_dot_recall@3": 0.9866666666666667, |
| "eval_dot_recall@5": 0.9866666666666667, |
| "eval_runtime": 0.2139, |
| "eval_samples_per_second": 0.0, |
| "eval_steps_per_second": 0.0, |
| "step": 3 |
| }, |
| { |
| "epoch": 4.0, |
| "eval_cosine_accuracy@1": 0.88, |
| "eval_cosine_accuracy@10": 1.0, |
| "eval_cosine_accuracy@3": 0.9866666666666667, |
| "eval_cosine_accuracy@5": 0.9866666666666667, |
| "eval_cosine_map@100": 0.9330158730158731, |
| "eval_cosine_mrr@10": 0.9330158730158731, |
| "eval_cosine_ndcg@10": 0.9499978881111136, |
| "eval_cosine_precision@1": 0.88, |
| "eval_cosine_precision@10": 0.09999999999999998, |
| "eval_cosine_precision@3": 0.3288888888888888, |
| "eval_cosine_precision@5": 0.1973333333333333, |
| "eval_cosine_recall@1": 0.88, |
| "eval_cosine_recall@10": 1.0, |
| "eval_cosine_recall@3": 0.9866666666666667, |
| "eval_cosine_recall@5": 0.9866666666666667, |
| "eval_dot_accuracy@1": 0.8933333333333333, |
| "eval_dot_accuracy@10": 1.0, |
| "eval_dot_accuracy@3": 0.9866666666666667, |
| "eval_dot_accuracy@5": 0.9866666666666667, |
| "eval_dot_map@100": 0.9394444444444444, |
| "eval_dot_mrr@10": 0.9394444444444445, |
| "eval_dot_ndcg@10": 0.9546805786428596, |
| "eval_dot_precision@1": 0.8933333333333333, |
| "eval_dot_precision@10": 0.09999999999999998, |
| "eval_dot_precision@3": 0.3288888888888888, |
| "eval_dot_precision@5": 0.1973333333333333, |
| "eval_dot_recall@1": 0.8933333333333333, |
| "eval_dot_recall@10": 1.0, |
| "eval_dot_recall@3": 0.9866666666666667, |
| "eval_dot_recall@5": 0.9866666666666667, |
| "eval_runtime": 0.2064, |
| "eval_samples_per_second": 0.0, |
| "eval_steps_per_second": 0.0, |
| "step": 4 |
| }, |
| { |
| "epoch": 5.0, |
| "eval_cosine_accuracy@1": 0.88, |
| "eval_cosine_accuracy@10": 1.0, |
| "eval_cosine_accuracy@3": 0.9866666666666667, |
| "eval_cosine_accuracy@5": 0.9866666666666667, |
| "eval_cosine_map@100": 0.9330158730158731, |
| "eval_cosine_mrr@10": 0.9330158730158731, |
| "eval_cosine_ndcg@10": 0.9499978881111136, |
| "eval_cosine_precision@1": 0.88, |
| "eval_cosine_precision@10": 0.09999999999999998, |
| "eval_cosine_precision@3": 0.3288888888888888, |
| "eval_cosine_precision@5": 0.1973333333333333, |
| "eval_cosine_recall@1": 0.88, |
| "eval_cosine_recall@10": 1.0, |
| "eval_cosine_recall@3": 0.9866666666666667, |
| "eval_cosine_recall@5": 0.9866666666666667, |
| "eval_dot_accuracy@1": 0.88, |
| "eval_dot_accuracy@10": 1.0, |
| "eval_dot_accuracy@3": 0.9866666666666667, |
| "eval_dot_accuracy@5": 0.9866666666666667, |
| "eval_dot_map@100": 0.9330158730158731, |
| "eval_dot_mrr@10": 0.9330158730158731, |
| "eval_dot_ndcg@10": 0.9499978881111136, |
| "eval_dot_precision@1": 0.88, |
| "eval_dot_precision@10": 0.09999999999999998, |
| "eval_dot_precision@3": 0.3288888888888888, |
| "eval_dot_precision@5": 0.1973333333333333, |
| "eval_dot_recall@1": 0.88, |
| "eval_dot_recall@10": 1.0, |
| "eval_dot_recall@3": 0.9866666666666667, |
| "eval_dot_recall@5": 0.9866666666666667, |
| "eval_runtime": 0.2216, |
| "eval_samples_per_second": 0.0, |
| "eval_steps_per_second": 0.0, |
| "step": 5 |
| } |
| ], |
| "logging_steps": 10, |
| "max_steps": 5, |
| "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": 0.0, |
| "train_batch_size": 32, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
