make exp3 works but no check

2025-12-06 13:10:02 +08:00
parent 1061780ea5
commit 45b60b269f
9 changed files with 254 additions and 72 deletions
--- a/exp2/modified/predict.py
+++ b/exp2/modified/predict.py
@@ -83,7 +83,7 @@ class Predictor:
        :param image: 该列表的shape必须为28x28。
        :return: 预测结果。
        """
-        input = torch.Tensor(image).float()
+        input = torch.tensor(image, dtype=torch.float32)
        assert(input.dim() == 2)
        assert(input.size(0) == 28)
        assert(input.size(1) == 28)