pytorch怎样导入神经网络的模型参数

 时间:2026-02-12 17:50:51

1、在github中可以很容易找到Resnet和VGG的网络结构,以pytorch中导入Resnet和VGG的神经网络模型的参数为例,其中左图为Resnet的网络结构,右图为VGG的网络结构。

pytorch怎样导入神经网络的模型参数

2、Resnet和VGG的网络结构训练的数据集是人脸表情数据集FER2013,下图所示为Resnet和VGG分类FER2013的参数模型。

pytorch怎样导入神经网络的模型参数

3、进入windows的cmd界面,把目录切换到模型参数导入的程序文件夹下,并激活带有pytorch的虚拟环境。

pytorch怎样导入神经网络的模型参数

4、如下图程序所示,引入VGG网络结构,并把VGG训练FER2013的模型参数导入的网络中,并且将参数导入到CUDA中,利用GPU进行加速计算。

pytorch怎样导入神经网络的模型参数

5、下图为导入VGG网络结构,以及FER2013_VGG的模型参数导入网络的输出结果。

pytorch怎样导入神经网络的模型参数

6、如下图程序所示,引入Resnet的网络结构,并把Resnet训练FER2013的模型参数导入的网络中,并且将参数导入到CUDA中,利用GPU进行加速计算。

pytorch怎样导入神经网络的模型参数

7、下图为导入Resnet网络结构,以及FER2013_Resnet的模型参数导入网络的输出结果。

pytorch怎样导入神经网络的模型参数

8、模型参数导入的最重要的命令为checkpoint = torch.load()和model.load_state_dict(checkpoint['net']),输出的结果如下所示,可以看到所有导入网络参数的具体数值。

pytorch怎样导入神经网络的模型参数

  • vlog用什么剪辑软件?
  • Selenium快速获得网页元素定位路径的方法
  • 生二胎,到底要不要和老大商量?
  • 如何招聘到优秀员工
  • 烟草花怎么画
  • 热门搜索
    尤格萨隆怎么打 花甲怎么洗 未就业证明怎么开 怎么读音标 惭愧的近义词 白衣服发黄怎么洗 适宜的近义词 三洋洗衣机怎么样 感谢的近义词 电脑怎么设置自动开机