release_summary.json

{
  "version": "v0.1.0",
  "repo": "diverWayne/cnn-cifar10-classifier",
  "github": "https://github.com/diverHansun/cnn-cifar10-classifier",
  "dataset": {
    "name": "uoft-cs/cifar10",
    "config": "plain_text",
    "split": "train/test"
  },
  "model": {
    "architecture": "SimpleCNN",
    "parameters": 620362,
    "classes": [
      "airplane",
      "automobile",
      "bird",
      "cat",
      "deer",
      "dog",
      "frog",
      "horse",
      "ship",
      "truck"
    ]
  },
  "training": {
    "epochs": 20,
    "batch_size": 256,
    "optimizer": "SGD",
    "learning_rate": 0.01,
    "momentum": 0.9,
    "weight_decay": 0.0005,
    "seed": 42,
    "amp": true,
    "augment": true
  },
  "environment": {
    "gpu": "NVIDIA GeForce RTX 5070 Ti",
    "torch": "2.11.0+cu128",
    "cuda": "12.8"
  },
  "metrics": {
    "train_loss": 0.6156930644035339,
    "train_accuracy": 0.7846,
    "test_loss": 0.6217316284179687,
    "test_accuracy_from_training": 0.7855,
    "test_accuracy_from_evaluate": 0.7857,
    "best_checkpoint_epoch": 19
  }
}