当前位置: 首页 > article >正文

每天五分钟深度学习框架pytorch:神经网络模型的参数初始化操作

本文重点

其实如果我们使用pytorch封装好的网络层的时候,我们并不需要对模型的参数初始化,因为这些都是pytorch帮助我们完成的,但是如果我们自己搭建模型(不使用pytorch中的封装好的网络层)或者对pytorch中封装好的模型初始化参数不满意,那么此时我们对模型进行参数初始化。

用pytorch已经封装好的层来搭建网络模型

import torch

import numpy as np

from torch import nn

class Simple_net(nn.Module):

def __init__(self):

super(Simple_net,self).__init__()

self.layer1=nn.Linear(4,3)

def forward(self,x):

return self.layer1(x)

net=Simple_net()

print(net)

print(net.layer1.weight.shape) #torch.Size([3, 4])

data=torch.randn(5,4)</


http://www.kler.cn/a/404185.html

相关文章:

  • [C++]:C++11(三)
  • Unity3d场景童话梦幻卡通Q版城镇建筑植物山石3D模型游戏美术素材
  • STM32H7开发笔记(2)——H7外设之多路定时器中断
  • 图形学笔记 - 4. 几何 -网格操作和阴影映射
  • 鸿蒙实战:使用隐式Want启动Ability
  • 现在转前端怎么样?
  • (二)Ubuntu22.04+Stable-Diffusion-webui AI绘画 中英双语插件安装
  • 括号匹配算法
  • Kafka-创建topic源码
  • Flink的Standalone集群模式安装部署
  • 【机器学习chp6】对数几率回归
  • 【AI】人工智能报告解读——中国人工智能的发展
  • #systemverilog# 关于 randomize(a) 却报 b 失败的疑问
  • pytorch经典训练流程
  • 【运维自动化-作业平台】如何使用全局变量之数组类型?
  • C#桌面应用制作计算器进阶版01
  • 空间与单细胞转录组学的整合定位肾损伤中上皮细胞与免疫细胞的相互作用
  • 稀疏最大谐波噪声比解卷积算法MATLAB实战
  • 十八:HTTP包体的传输方式(1):定长包体
  • 如何删除pdf里的任意一页?删除PDF里任意一页的几种方法
  • 算法知识-13-链表
  • 蓝桥杯每日真题 - 第20天
  • 「Mac玩转仓颉内测版25」基础篇5 - 布尔类型详解
  • HTTP 协议的作用
  • [开源] SafeLine 好用的Web 应用防火墙(WAF)
  • Java Swing-1.基本概念及组件