上传者: 38694023
|
上传时间: 2021-12-28 17:16:58
|
文件大小: 47KB
|
文件类型: -
听说pytorch使用比TensorFlow简单,加之pytorch现已支持windows,所以今天装了pytorch玩玩,第一件事还是写了个简单的CNN在MNIST上实验,初步体验的确比TensorFlow方便。
参考代码(在莫烦python的教程代码基础上修改)如下:
import torch
import torch.nn as nn
from torch.autograd import Variable
import torch.utils.data as Data
import torchvision
import time
#import matplotlib.pyplot a