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

 时间:2024-10-27 07:54:05

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&#泌驾台佐39;]),输出的结果如下所示,可以看到所有导入网络参数的具体数值。

pytorch怎样导入神经网络的模型参数
  • vlog用什么剪辑软件?
  • Selenium快速获得网页元素定位路径的方法
  • 生二胎,到底要不要和老大商量?
  • 如何招聘到优秀员工
  • 烟草花怎么画
  • 热门搜索
    防火手抄报图片大全 重阳节最漂亮手抄报 法制安全手抄报内容 中国神话故事手抄报 国庆小报手抄报 快乐读书手抄报图片 三年级国庆节手抄报 文明交通手抄报图片 走进名著手抄报内容 数学手抄报的资料