Pytorch数据集下载触发IDM

bejyjqdl  于 2023-03-30  发布在  其他
关注(0)|答案(1)|浏览(126)

我通常使用Internet Download Manager来下载文件,但我不希望Pytorch触发IDM所有我尝试下载的数据集。例如:

training_data = datasets.FashionMNIST(
    root="data",
    train=True,
    download=True,
    transform=ToTensor()
)

触发IDM,如下所示:

我有办法让IDM不这样做吗?这是IDM的错吗?

7lrncoxx

7lrncoxx1#

您可以通过以下方式将进度条设置为PyTorch内置的下载进度条:

import os
os.environ['TORCH_HOME'] = './data'

相关问题