当前位置: 代码迷 >> 综合 >> 训练模型的时候, Warning: NaN or Inf found in input tensor 解决办法
  详细解决方案

训练模型的时候, Warning: NaN or Inf found in input tensor 解决办法

热度:61   发布时间:2023-12-15 00:46:36.0

一般出现这句话的时候表示,模型训练过程中出现梯度消失或者梯度爆炸。 

更改方式是降低初始学习率,比如加入初始学习率是0.1那么就将它改成0.01。

 

 

  相关解决方案