spicecloud
/

spice-cnn-base

Image Classification

Model card Files Files and versions

rhendz commited on Jun 13, 2023

Commit

4533e20

·

1 Parent(s): dcd84f0

Upload model

Files changed (2) hide show

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

modeling_spice_cnn.py CHANGED Viewed

@@ -37,7 +37,6 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
             nn.Dropout(0.4),
             nn.Flatten(),
-            nn.Linear(14*14*5, 128),
             nn.BatchNorm1d(128),
             nn.ReLU(),
             nn.Dropout(0.4),

             nn.Dropout(0.4),
             nn.Flatten(),
             nn.BatchNorm1d(128),
             nn.ReLU(),
             nn.Dropout(0.4),

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:13d74c7b24138f2a238170dceb0626116f1c7a572d91ce364530e8945482b2c8
+size 796341