摘要: 一.MNIST 数据集分类 1.加载数据 2.显示数据集中的部分图像 3.创建网络 定义网络时,需要继承nn.Module,并实现它的forward方法,把网络中具有可学习参数的层放在构造函数init中。 只要在nn.Module的子类中定义了forward函数,backward函数就会自动被实现( 阅读全文
posted @ 2021-10-16 19:52 浮沉& 阅读(149) 评论(0) 推荐(0)