首页 > 其他分享 >pytorch net.state_dict问题

pytorch net.state_dict问题

时间:2023-02-02 23:55:04浏览次数:38  
标签:__ nn self state pytorch dict net model

pytorch使用中发现net.state_dict只保存所有module层的偏置与权重值,不保存零散变量值。
验证代码

#encoding:utf-8
 
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision
import numpy as mp
import matplotlib.pyplot as plt
import torch.nn.functional as F
 
#define model
class TheModelClass(nn.Module):
    def __init__(self):
        super(TheModelClass,self).__init__()
        self.conv1=nn.Conv2d(3,6,5)
        self.pool=nn.MaxPool2d(2,2)
        self.conv2=nn.Conv2d(6,16,5)
        self.fc1=nn.Linear(16*5*5,120)
        self.fc2=nn.Linear(120,84)
        self.fc3=nn.Linear(84,10)
        self.qx=0
 
    def forward(self,x):
        x=self.pool(F.relu(self.conv1(x)))
        x=self.pool(F.relu(self.conv2(x)))
        x=x.view(-1,16*5*5)
        x=F.relu(self.fc1(x))
        x=F.relu(self.fc2(x))
        x=self.fc3(x)
        return x
 
def main():
    # Initialize model
    model = TheModelClass()
 
    #Initialize optimizer
    optimizer=optim.SGD(model.parameters(),lr=0.001,momentum=0.9)
 
    #print model's state_dict
    print('Model.state_dict:')
    for param_tensor in model.state_dict():
        #打印 key value字典
        print(param_tensor,'\t',model.state_dict()[param_tensor].size())
 
    #print optimizer's state_dict
    print('Optimizer,s state_dict:')
    for var_name in optimizer.state_dict():
        print(var_name,'\t',optimizer.state_dict()[var_name])
if __name__=='__main__':
    main()

model的state_dict输出中并没有qx变量值。

标签:__,nn,self,state,pytorch,dict,net,model
From: https://www.cnblogs.com/xzh-personal-issue/p/17087793.html

相关文章

  • Net6 ConfigurationBinder configuration转为对象(未完成)
    绑定集合varsource=newDictionary<string,string>{["0:gender"]="Male",//["0:age"]="18",//["0:contactInfo:emailAddress"]="[email protected]",//["0......
  • 一个强大、支持100多种格式.Net图片操作库
    更多开源项目请查看:一个专注推荐优秀.Net开源项目的榜单我们在项目开发中,对图片的操作可以说是非常常见,比如图片大小改变、图片合并、格式转换、添加水印等,自己开发都需......
  • VS 禁止自动创建平台子文件夹“net48”
    问题:最近在创建vb.net或者c#项目的时候,总是会在设定的输出路径下自动创建一个“net48”的平台文件夹,并将dll生成在这个文件夹里面,而调试启动的exe在设定的输出路径下,导致调......
  • kubetpl - kubernetes 模板管理工具
    目录Helm、Kustomize、KubetplHelmKustomizeKubetpl安装KubetplKubetpl命令参数参数选项completion-参数自动补齐render-渲染模板go-template语法注释引用变量在te......
  • kubernetes-dashboard 实现 http 访问以及免 token 登录
    目录下载官方yaml文件修改yaml文件修改service端口修改clusterrolebinding修改deployment内容修改探针检测修改镜像拉取策略修改容器端口关闭token登录增加ing......
  • Efficient training of physics-informed neural networks via importance sampling
     因为看着作者是英伟达的,便看了一下。总体感觉没有什么新意,改进幅度也很小,但是理论推导可以看一下。可以借鉴一下。本文通过重要性采样对PINN进行高效计算,本文提出的......
  • kubernetes关于eks一次异常问题的复盘
    背景:海外新加坡有一套aws的eks集群,很小的规模托管的三节点(172-31-16-189节点为最近才加的,忽略):[root@ip-172-31-10-1~]#kubectlgetnodesNAME......
  • kubernetes(四)
    一、基于StatefulSet部署有状态访问、基于DaemonSet在每一个node节点部署一个prometheusnode-exporter1.1StatefulSethttps://kubernetes.io/zh-cn/docs/concepts/work......
  • .net7 htt2.0 htt3.0
    服务器端代码配置varbuilder=WebApplication.CreateBuilder(args);builder.WebHost.ConfigureKestrel((context,options)=>{......
  • Quartz.Net源码Example之Quartz.Examples
    Quartz.Examples反射-Example批量执行​ 实现思路:定义一个统一的接口,需要实现的类全部实现该接口;通过反射获取实现该接口的实例并触发其中的方法。定义统一的接口//......