Upload model

modeling_spice_cnn.py

@@ -39,6 +39,7 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
     def forward(self, tensor, labels=None):
         logits = self.model(tensor)
         if labels is not None:
-            loss = nn.cross_entropy(logits, labels)
+            loss_fnc = nn.CrossEntropyLoss()
+            loss = loss_fnc(logits, labels)
             return {"loss": loss, "logits": logits}
         return {"logits": logits}

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:661a29cd1981046365bd851e3c2a001dfe0732bd9332065350ea3bb31eb4c1a9
+oid sha256:ba7c626aead32b62f406aea958f517e0c6e01afcd6ff18b5931da0cfc2325e31
 size 1691531