Implementing and Training a LeNet-5 Variant with PyTorch
Model Definition
File: model.py
import torch
import torch.nn as nn
from torchsummary import summary
class LeNet5Variant(nn.Module):
def __init__(self, input_channels=1, num_classes=10):
super().__init__()
# First convolutional block
self.conv_block_1 = nn.Sequential(
nn.Conv2d(input_channels, 6, kernel_ ...
Posted on Sat, 07 Nov 2026 16:05:14 +0000 by saandel