pytorch-mirror 0.5.0

pytorch模型、数据集国内镜像

PyTorch模型、数据集国内镜像

一、使用方法

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

import torch
import torch_dataset_mirror
import torch_model_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_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