g0401828t/LeNet

LeNet implemented by pytorch

LeNet

LeNet implemented with pytorch

• main.py에서 parameter 부분의 path (경로)만 본인에 맞게 수정 필요

C1

nn.Conv2d(1, 6, k=5, stride=1)

S2

subsampling F.avg_pool2d(input, k=2, stride=2)

C3

nn.Conv2d(6, 16, k=5, stride=1)

S4

subsampling F.avg_pool2d(input, k=2, stride=2)

C5

nn.Conv2d(16, 120, k=5, stride=1)

F4

fully connected layers nn.Linear(120, 84) nn.Linear(84, 10) F.softmax()