加载本地已下载的CIFAR数据集报错解决办法

加载本地已下载的CIFAR数据集报错解决办法

使用torchvision.datasets加载数据集时,本地下载好数据集后,将所有程序文件打包放别的机器上运行时,设置不下载,发现仍然需要重复下载数据集

train_data = torchvision.datasets.CIFAR10(root='../data',train=True,transform=transform,download=False)
test_data = torchvision.datasets.CIFAR10(root='../data',train=False,transform=transform,download=False)```

```bash
RuntimeError: Dataset not found or corrupted. You can use download=True to download it

解决方法:使用下载好的数据集的绝对路径

train_data = torchvision.datasets.CIFAR10(root='/home/file/data/',train=True,transform=transform,download=False)
test_data = torchvision.datasets.CIFAR10(root='/home/file/data/',train=False,transform=transform,download=False)

你可能感兴趣的:(数据集加载,深度学习,pytorch,tensorflow)