Upload model

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ab5a7220d6d3c95d15f9138ce25cc3c95b569c0d056e60893d5f28c5f50a2325
+oid sha256:1afa15c915cfba02cf65f9fda685e845a333113fdf807050238d1d8136c8a7ec
 size 247728

modeling_lenet.py

@@ -11,18 +11,18 @@ class LeNet(torch.nn.Module):
         self.model = torch.nn.Sequential(
             OrderedDict(
                 [
-                    ("conv1", torch.nn.LazyConv2d(6, 5, padding=2)),
+                    ("conv1", torch.nn.Conv2d(1, 6, 5, padding=2)),
                     ("pooling1", torch.nn.AvgPool2d(2, stride=2)),
                     ("sigmoid1", torch.nn.Sigmoid()),
-                    ("conv2", torch.nn.LazyConv2d(16, 5)),
+                    ("conv2", torch.nn.Conv2d(6, 16, 5)),
                     ("pooling2", torch.nn.AvgPool2d(2, stride=2)),
                     ("sigmoid1", torch.nn.Sigmoid()),
                     ("flatten", torch.nn.Flatten()),
-                    ("dense1", torch.nn.LazyLinear(120)),
+                    ("dense1", torch.nn.Linear(400, 120)),
                     ("sigmoid3", torch.nn.Sigmoid()),
-                    ("dense2", torch.nn.LazyLinear(84)),
+                    ("dense2", torch.nn.Linear(120, 84)),
                     ("sigmoid4", torch.nn.Sigmoid()),
-                    ("dense3", torch.nn.LazyLinear(10)),
+                    ("dense3", torch.nn.Linear(84, 10)),
                 ]
             )
         )