Upload model

modeling_spice_cnn.py

@@ -39,7 +39,6 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
     def forward(self, tensor, labels=None):
         logits = self.model(tensor)
         if labels is not None:
-            print(labels, labels.shape)
-            loss = nn.CrossEntropyLoss(logits, labels)
+            loss = nn.cross_entropy(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:01320bde65455e7861ab8ed732fe4d5bfa0cd67de4c50a922112d5b4dbcb910e
+oid sha256:661a29cd1981046365bd851e3c2a001dfe0732bd9332065350ea3bb31eb4c1a9
 size 1691531