pytorch datasets.CIFAR10保存在当前路径
时间: 2023-10-16 21:09:43 浏览: 187
DPN-CIFAR10
如果你使用的是 Jupyter Notebook 或者 Colab,可以使用以下代码进行下载和解压:
```python
import torchvision
from torchvision import datasets
# 下载CIFAR10
trainset = datasets.CIFAR10(root='./', train=True, download=True)
testset = datasets.CIFAR10(root='./', train=False, download=True)
```
如果你使用的是命令行或者 Python 脚本,可以使用以下代码进行下载和解压:
```python
import torchvision
from torchvision import datasets
# 下载CIFAR10
trainset = datasets.CIFAR10(root='./', train=True, download=True)
testset = datasets.CIFAR10(root='./', train=False, download=True)
```
下载完成后,CIFAR10 数据集会被保存在当前路径下。你可以使用 `trainset` 和 `testset` 访问训练集和测试集的数据。
阅读全文