解决CrossEntropyLoss测试因为数值问题无法通过测试的问题

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")