【pytorch22】激活函数与GPU加速

激活函数


ReLu还是还是可能出现梯度弥散,因为x<0的时候,梯度还是可能小于0

leaky relu,在x<0的时候,梯度就不会为0,梯度不会不动

Relu函数在x=0处是不连续的

一种更加光滑的曲线是SELU,是两个函数的concat

softplus,把ReLu的尖点x=0做了一个平滑处理,使得导数不连续的地方导数连续

目前用的最多的是Relu、sigmoid,tan h(RNN)以及Leaky Relu

GPU加速

loss层也是需要计算的,可以把loss层也搬到GPU上去

相关推荐
晚霞的不甘1 天前
CANN 模型转换与适配:从 PyTorch 到 Ascend OM 的完整指南
人工智能·pytorch·python·深度学习
speop1 天前
【thorough-pytorch】评价指标
人工智能·pytorch·python
晚霞的不甘1 天前
CANN-昇腾NPU开发快速入门
人工智能·pytorch·python·深度学习
rayyy91 天前
神经网络拟合高频信号实验
人工智能·pytorch·神经网络
香蕉鼠片1 天前
CUDA、PyTorch、Transformers、PEFT 全栈详解
人工智能·pytorch·python
MediaTea1 天前
PyTorch:张量与基础计算模块
人工智能·pytorch·python·深度学习·机器学习
放下华子我只抽RuiKe52 天前
React 从入门到生产(一):JSX 与组件思维
前端·javascript·人工智能·pytorch·深度学习·react.js·前端框架
z小猫不吃鱼2 天前
05 Pytorch之 ViT-B/16 源码逐行解析
人工智能·pytorch·python
xG8XPvV5d2 天前
PyTorch特征提取器源码精析
人工智能·pytorch·python