spicecloud
/

spice-cnn-base

Image Classification

Model card Files Files and versions

rhendz commited on Jun 13, 2023

Commit

aa4a9d0

·

1 Parent(s): 6f57bec

Upload model

Files changed (2) hide show

modeling_spice_cnn.py +1 -1
pytorch_model.bin +2 -2

modeling_spice_cnn.py CHANGED Viewed

@@ -26,7 +26,7 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
             nn.ReLU(),
             nn.MaxPool2d(kernel_size=config.pooling_size),
-            nn.Linear(64*3*3, 128),
             nn.ReLU(),
             nn.Dropout(0.5),
             nn.Linear(128, config.num_classes)

             nn.ReLU(),
             nn.MaxPool2d(kernel_size=config.pooling_size),
+            nn.Linear(64*2*2, 128),
             nn.ReLU(),
             nn.Dropout(0.5),
             nn.Linear(128, config.num_classes)

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:27bc8f3fbf442d90430939e9d7d9569b01ddb1f98c7fd9a026784d6f01471b55
+size 238972