pytorch-mirror 0.5.1

pytorch模型、数据集国内镜像

PyTorch模型、数据集国内镜像

一、使用方法

只需要在使用模型或数据集之前导入torchmirror.dataset或torchmirror.model模块即可，如下：

import torch
import torchvision.datasets.mnist as mnist
from torchmirror import dataset, model

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

二、使用其他镜像

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

export PYTORCH_MODEL_MIRROR=http://192.168.0.100:8080
export PYTORCH_DATASET_MIRROR=http://192.168.0.100:8080