diff --git a/test/core/test_loss.py b/test/core/test_loss.py
index a57e6542..976285a9 100644
--- a/test/core/test_loss.py
+++ b/test/core/test_loss.py
@@ -18,13 +18,13 @@ class TestLoss(unittest.TestCase):
         a = torch.randn(3, 4, 3)
         b = torch.randint(3, (3, 3))
         ans = ce({"my_predict": a}, {"my_truth": b})
-        self.assertEqual(ans, torch.nn.functional.cross_entropy(a, b))
+        self.assertAlmostEqual(ans.item(), torch.nn.functional.cross_entropy(a, b).item(), places=4)
 
         ce = loss.CrossEntropyLoss(pred="my_predict", target="my_truth", class_in_dim=2)
         a = torch.randn(3, 4, 3)
         b = torch.randint(3, (3, 4))
         ans = ce({"my_predict": a}, {"my_truth": b})
-        self.assertEqual(ans, torch.nn.functional.cross_entropy(a.transpose(1, 2), b))
+        self.assertAlmostEqual(ans.item(), torch.nn.functional.cross_entropy(a.transpose(1, 2), b).item(), places=4)
     
     def test_BCELoss(self):
         bce = loss.BCELoss(pred="my_predict", target="my_truth")