Search code examples
pytorch

Pytorch dataset download triggers IDM


I normally use Internet Download Manager to download my files, but I don't want Pytorch triggering IDM everything I try to download datasets. For example:

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

Triggers IDM like this: enter image description here

Is there anyway for me to make IDM not do this? Is this IDM's fault?


Solution

  • You can set the progress bar to the PyTorch's built-in download progress bar by:

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