问题描述

报错：ConnectionResetError: [WinError 10054] 远程主机强迫关闭了一个现有的连接..........

 原因分析

这是因为我们连接失败之后系统反复请求连接，被判定为恶意访问，中断连接。

解决方案

步骤1：

ctrl+鼠标左键  点击图一上方的红色方框内的"Downloading:"后面的网址，浏览器会开始下载，相当于我们通过浏览器预先下载好了所需要的.pth文件，而不用在终端自动下载。

步骤2：

在调用模型的时候使用下面代码进行加载。

import torch
from torchvision import models
 
 
# 加载模型，以DenseNet为例
model_ft = models.densenet121(pretrained=False)
# 通过网址下载下来的.pth文件全称
preload = torch.load("densenet121-a639ec97.pth")
# 加载完毕
model_ft.load_state_dict(preload)

希望我的经验分享能够帮到您，如果觉得写的不错可以点赞三连多多支持哦，谢谢您的观看！！！