Upload processor
Browse files- preprocessor_lenet.py +1 -1
preprocessor_lenet.py
CHANGED
|
@@ -28,7 +28,7 @@ class LeNetProcessor(BaseImageProcessor):
|
|
| 28 |
images = [images]
|
| 29 |
|
| 30 |
transform = v2.Compose([
|
| 31 |
-
v2.
|
| 32 |
v2.ToDtype(torch.float32, scale=True),
|
| 33 |
v2.Normalize(
|
| 34 |
mean=[0.1307],
|
|
|
|
| 28 |
images = [images]
|
| 29 |
|
| 30 |
transform = v2.Compose([
|
| 31 |
+
v2.Resize(size=(28, 28), antialias=True),
|
| 32 |
v2.ToDtype(torch.float32, scale=True),
|
| 33 |
v2.Normalize(
|
| 34 |
mean=[0.1307],
|