training_info.json

{
  "arch": "convnext_tiny",
  "pretrained_weights": "IMAGENET1K_V1",
  "dataset": "tanganke/stanford_cars",
  "label_mode": "brand",
  "num_classes": 49,
  "class_names": [
    "AM General",
    "Acura",
    "Aston Martin",
    "Audi",
    "BMW",
    "Bentley",
    "Bugatti",
    "Buick",
    "Cadillac",
    "Chevrolet",
    "Chrysler",
    "Daewoo",
    "Dodge",
    "Eagle",
    "FIAT",
    "Ferrari",
    "Fisker",
    "Ford",
    "GMC",
    "Geo",
    "HUMMER",
    "Honda",
    "Hyundai",
    "Infiniti",
    "Isuzu",
    "Jaguar",
    "Jeep",
    "Lamborghini",
    "Land Rover",
    "Lincoln",
    "MINI",
    "Maybach",
    "Mazda",
    "McLaren",
    "Mercedes-Benz",
    "Mitsubishi",
    "Nissan",
    "Plymouth",
    "Porsche",
    "Ram",
    "Rolls-Royce",
    "Scion",
    "Spyker",
    "Suzuki",
    "Tesla",
    "Toyota",
    "Volkswagen",
    "Volvo",
    "smart"
  ],
  "epochs": 8,
  "image_size": 288,
  "batch_size": 16,
  "backbone_lr": 0.0001,
  "head_lr": 0.001,
  "majority_baseline": 0.11112475442043222,
  "val_accuracy": 0.9202453987730062,
  "test_accuracy_plain": 0.9241387887078721,
  "test_accuracy_tta": 0.9269991294615098,
  "previous_model_accuracy": 0.8514,
  "history": {
    "train_loss": [
      2.783953580273291,
      0.9787385308846636,
      0.5814701036334559,
      0.4797711311859855,
      0.43764293460606485,
      0.4189803975768485,
      0.4121319403684816,
      0.4095040149173362
    ],
    "train_acc": [
      0.30472161572052403,
      0.8332423580786026,
      0.9596069868995634,
      0.9892194323144105,
      0.996042576419214,
      0.9987718340611353,
      0.9995906113537117,
      0.9991812227074236
    ],
    "val_loss": [
      1.774648526548608,
      0.9960233030875036,
      0.8324032773269466,
      0.7850571597280678,
      0.7175895720171782,
      0.7131279548984364,
      0.6997977729955334,
      0.6970531023353156
    ],
    "val_acc": [
      0.5717791411042945,
      0.8171779141104294,
      0.8748466257668711,
      0.8957055214723927,
      0.9141104294478528,
      0.9141104294478528,
      0.9165644171779141,
      0.9202453987730062
    ]
  }
}