Upload 2 files

butterfly_resnet_checkpoint.bin

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

config.json

@@ -0,0 +1,15 @@
+{
+    "model_name": "SoulPerforms/Butterfly_image_classification_resnet18",
+    "architecture": "resnet18",
+    "num_classes": 10,
+    "input_size": [3, 224, 224],
+    "batch_size": 32,
+    "learning_rate": 0.001,
+    "epochs": 10,
+    "optimizer": "Adam",
+    "loss_function": "CrossEntropyLoss",
+    "pretrained": true,
+    "dropout": 0.5,
+    "seed": 42,
+    "device": "cuda"  // or "cpu"
+}