torchvision mnist RemoteDisconnected：远端关闭连接无响应

Question

我有一个 pytorch 和 mnist 错误...... 为什么会出现这个错误？

RemoteDisconnected：远端关闭连接无响应

    import torch
    import torchvision.datasets as dsets
    import torchvision.transforms as transforms
    import matplotlib.pyplot as plt
    import random

    training_epochs = 15
    batch_size = 100

    'mnist_train = dsets.MNIST(root='MNIST_data/',
                              train=True,
                              transform=transforms.ToTensor(),
                              download=True)'

    mnist_test = dsets.MNIST(root='MNIST_data/',
                             train=False,
                             transform=transforms.ToTensor(),
                             download=True)

---------------------------------------------------------------------------------

RemoteDisconnected                        Traceback (most recent call last)
c:\Users\PSG\Desktop\Code\Python\pytorch\lab-07_2_mnist_introduction.py in 
     22                           train=True,
     23                           transform=transforms.ToTensor(),
---> 24                           download=True)
     25 
     26 mnist_test = dsets.MNIST(root='MNIST_data/',

RemoteDisconnected: Remote end closed connection without response

---------------------------------------------------------------------------------

Answer 1

当我运行此代码时：

import torch
model = torch.hub.load('mateuszbuda/brain-segmentation-pytorch', 'unet',
    in_channels=3, out_channels=1, init_features=32, pretrained=True)

我遇到这样的错误：

http.client.RemoteDisconnected：远端关闭连接无响应

有多种方法可以解决这个问题。请注意一些提示。

当我运行上面的代码时，我得到了如下建议：

将命令更改为 {calling_fn}(..., trust_repo=False)

或

您还可以使用 load(..., trust_repo='check')

然后我将参数添加到我的代码中：

model = torch.hub.load('mateuszbuda/brain-segmentation-pytorch', 'unet',
        in_channels=3, out_channels=1, init_features=32, pretrained=True, trust_repo='check')

之后代码运行良好。

此外，您还可以获取正在从命令行下载的文件的 URL 引用，例如：

下载：“https://github.com/mateuszbuda/brain-segmentation-pytorch/zipball/master”到/home/xxx/.cache/torch/hub/master.zip

因此，您可以手动下载文件。