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

 时间:2026-05-04 12:18:13

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快速获得网页元素定位路径的方法
  • 生二胎,到底要不要和老大商量?
  • 如何招聘到优秀员工
  • 烟草花怎么画
  • 热门搜索
    四年级科普手抄报 预防新型冠状病毒的手抄报 孝敬父母手抄报内容 关于校园的手抄报 关于祖国在我心中的手抄报 讲普通话 写规范字手抄报 关于钓鱼岛的手抄报 幸福班级手抄报 感恩父母手抄报简单 关于环保手抄报的内容