pytorch模型、数据集国内镜像
Project description
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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torchmirror-0.6.0.tar.gz
(4.3 kB
view hashes)
Built Distribution
Close
Hashes for torchmirror-0.6.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 856bfb0641de13b0c8e37f8ddfb3fd44a422779717a21def0ef9ddd8921bf8b5 |
|
MD5 | 6f7c431339263a1355b95e093c01a868 |
|
BLAKE2b-256 | 164a915c30d5a9cf4528442506885f04f8289c1974940476ea19c47792e8f1fc |