training_metadata.json

{
  "model_name": "google/gemma-2b",
  "training_time_seconds": 131.95324730873108,
  "num_train_samples": 180,
  "num_val_samples": 20,
  "max_length": 128,
  "seed": 42,
  "use_small_model": false,
  "final_metrics": {
    "train_runtime": 131.7045,
    "train_samples_per_second": 4.1,
    "train_steps_per_second": 0.251,
    "total_flos": 803663719170048.0,
    "train_loss": 0.31323031223181524,
    "epoch": 2.9333333333333336,
    "step": 33
  }
}