best_config.json

{
  "config": {
    "beta": 1.0,
    "batch_size": 4,
    "gradient_accumulation_steps": 4,
    "num_epochs": 3,
    "learning_rate": 1e-05
  },
  "metrics": {
    "train_runtime": 651.1233,
    "train_samples_per_second": 4.607,
    "train_steps_per_second": 0.286,
    "total_flos": 0.0,
    "train_loss": 0.0037342177758611767,
    "epoch": 2.96
  },
  "dataset_params": {
    "max_examples": 500,
    "example_source": [
      0
    ]
  }
}