torch.nn.Sequential
是一个Sequential容器,模块将按照**构造函数中传递的顺序添加到模块中。**另外,也可以传入一个有序模块。
model = nn.Sequential(
nn.Conv2d(1,20,5),
nn.ReLU(),
nn.Conv2d(20,64,5),
nn.ReLU()
)
transforms.Compose
Compose里面的参数实际上就是个列表,而这个列表里面的元素就是你想要执行的transform操作。
transforms.Compose([
transforms.CenterCrop(10),
transforms.ToTensor(),])