在mnist上使用torchvision的架构

torchvision上的模型都是基于ImageNet的，想将其用于MNIST也很简单，只需要一点点修改(以resnet18为例)：

model = tv.models.resnet18(num_classes = 10)
model.conv1 = Conv2d(1, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False) #因为mnist是单通道的，而ImageNet是3通道

end

这个打钱就像赞一样，如果你想赞，可以赞一分

• Post author: fly理
• Post link: https://flyleeee.github.io/2021/09/21/post-5/
• Copyright Notice: All articles in this blog are licensed under unless otherwise stated.