fix: bad type hint and typo for forward method return value

2026-04-05 01:44:10 +02:00
parent 5a8fb2c48b
commit 82d61dd10f
2 changed files with 2 additions and 3 deletions
--- a/mnist_test.py
+++ b/mnist_test.py
@@ -29,7 +29,6 @@ def mnist_train(
    x_test.resize(x_test.shape[0], in_len)
    x_train = x_train / 255
    x_test = x_test / 255
-    x_train = x_train[:5000]
    if os.path.exists(filename):
        autoencoder = cls.load(filename)
    else: