| { | |
| "model_name": "distilbert-base-uncased", | |
| "num_epochs": 15, | |
| "batch_size": 16, | |
| "learning_rate": 3e-05, | |
| "train_samples": 659, | |
| "val_samples": 141, | |
| "final_metrics": { | |
| "eval_loss": 0.32609713077545166, | |
| "eval_accuracy": 0.9290780141843972, | |
| "eval_f1": 0.9268291188191063, | |
| "eval_precision": 0.9317931083888531, | |
| "eval_recall": 0.9290780141843972, | |
| "eval_runtime": 0.4923, | |
| "eval_samples_per_second": 286.387, | |
| "eval_steps_per_second": 18.28, | |
| "epoch": 10.0 | |
| }, | |
| "total_params": 66625416, | |
| "trainable_params": 66625416 | |
| } |