![](/img/trans.png)
[英]How to use MSELoss function for Fashion_MNIST in pytorch?
[英]How do I separate the input and targets from Pytorch Fashion MNIST?
Fashion MNIST 数据集在 Pytorch 中的实现非常奇怪。 我想做类似的事情:
X, y = FashionMNIST
但实际上,情况要复杂一些。 这就是我所拥有的:
from torchvision.datasets import FashionMNIST
train = FashionMNIST(root='.', download=True, train=True)
print(train)
输出:
Dataset FashionMNIST
Number of datapoints: 60000
Root location: c:/users/nicolas/documents/data/fashionmnist
Split: Train
一种观察结果如下:
print(train[0])
(<PIL.Image.Image image mode=L size=28x28 at 0x20868074780>, 9)
我只能做一次观察。
X, y = train[0]
那么如何分离输入和目标呢?
FashionMNIST
对象具有data
和targets
属性。
你可以简单地写
X, y = train.data, train.targets
然后你可以看到形状
X.shape, y.shape
(torch.Size([60000, 28, 28]), torch.Size([60000]))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.