Initial commit for ResNet-20 model

README.md

@@ -0,0 +1,11 @@
+# ResNet-20 model for CIFAR-10
+
+This is the repo for a custom-made neural network based on ResNet architecture that has been trained on CIFAR-10 dataset.
+
+## Model details
+
+- **Architecture:** ResNet
+- **Input shape:** 3x32x32
+- **Output classes:** 10
+- **Dataset:** CIFAR-10
+> More details can be found in the `config.json` file inside this repository.

config.json

@@ -0,0 +1,56 @@
+{
+     "model_architecture": {
+          "input_size": [
+               3,
+               32,
+               32
+          ],
+          "num_layers": 20,
+          "num_classes": 10,
+          "architecture": "ResNet"
+     },
+     "training_parameters": {
+          "learning_rate": 0.0625,
+          "batch_size": 128,
+          "num_epochs": 15,
+          "optimizer": "SGD",
+          "loss_function": "CrossEntropyLoss"
+     },
+     "dataset": {
+          "name": "CIFAR-10",
+          "normalization": {
+               "mean": [
+                    0.4914,
+                    0.4822,
+                    0.4465
+               ],
+               "std": [
+                    0.247,
+                    0.243,
+                    0.261
+               ]
+          },
+          "data_augmentation": [
+               "RandomCrop(32, padding=4)",
+               "RandomHorizontalFlip()"
+          ],
+          "num_examples": {
+               "train": 50000,
+               "valid": 9000,
+               "test": 1000
+          }
+     },
+     "performance": {
+          "accuracy": {
+               "train": 0.8683,
+               "valid": 0.8393,
+               "test": 0.848
+          }
+     },
+     "metadata": {
+          "author": "Sergey Polivin",
+          "date_trained": "2025-01-15",
+          "version": "v1.0",
+          "purpose": "Image classification on CIFAR-10"
+     }
+}

resnet20_weights.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:518cabc4124a21a5f16dd0e6e7a4ccb156c4cccf46378c36034e5686844056e1
+size 1139350