torch-dataset-mirror 0.2.1

pytorch数据集国内镜像

PyTorch数据集国内镜像

一、使用方法

只需要在使用数据集之前导入torch_dataset_mirror模块即可，如下：

import torch
import torch_dataset_mirror
import torchvision.datasets.mnist as mnist

train_dataset = mnist.MNIST("./data", train=True, download=True)
test_dataset = mnist.MNIST("./data", train=False, download=True)

二、使用其他镜像

要使用其他镜像，只需要修改PYTORCH_DATASET_MIRROR环境变量即可，如下：

export PYTORCH_DATASET_MIRROR=http://192.168.0.100:8080