torchvirsion下载数据集

53清

代码

1. %matplotlib inline
   
2. import torch
   
3. import torchvision
   
4. from torch.utils import data
   
5. from torchvision import transforms
   
6. from d2l import torch as d2l
   
7. 
   
8. 
   
9. d2l.use_svg_display
   
10. trans = transforms.ToTensor
    
11. 
    
12. mnist_train=torchvision.datasets.FashionMNIST
    
13. (
    
14.     root='./data',train=True,transform=trans,download=True
    
15.     )
    
16. mnist_train=torchvision.datasets.FashionMNIST
    
17. (
    
18.     root='./data',train=False,transform=trans,download=True
    
19.     )
    
20. len(mnist_train),len(mnist_test)
    
21. 
    
22. mnist_train[0][0].shape

复制代码

运行bug

53清

格式问题，把空格去掉就好