摘要: 直接贴代码: 首先导入相应的包: import torch import torchvision import torchvision.transforms as transforms import torch.nn as nn import torch.optim as optim from to 阅读全文
posted @ 2020-03-21 23:22 liualex_sone 阅读(416) 评论(0) 推荐(0) 编辑
摘要: 卷积计算公式:N = (W − F + 2P )/S+1,其中W指输入图片大小,F指卷积核大小,P指padding,S指stride 结果:由于只迭代了100次,结果只有78.57%。 阅读全文
posted @ 2019-08-16 19:50 liualex_sone 阅读(1407) 评论(0) 推荐(0) 编辑
摘要: 环境: pytorch1.1 cuda9.0 ubuntu16.04 该网络有3层,第一层input layer,有784个神经元(MNIST数据集是28*28的单通道图片,故有784个神经元)。第二层为hidden_layer,设置为500个神经元。最后一层是输出层,有10个神经元(10分类任务) 阅读全文
posted @ 2019-08-14 22:47 liualex_sone 阅读(5212) 评论(3) 推荐(0) 编辑