用GPU分别采用pytorch和tensorflow跑MNIST(LeNet-5模型)【可初步了解tnesorflow与pytorch转换】
pytorch版:总共分为几个步骤:1.加载数据集(这里当然可以用你自己的数据集,但要注意调节参数,h,w,channel)2.定义神经网络模型(这里写了两种方法)3.检查模型4.将模型传入gpu5.损失函数和优化器6.训练7.绘制曲线8.评估模型#LeNet-5网络结构importtorch.nnasnnimporttorchimporttorchvisionfromtorchvisionimp