2020-04-01 训练时出现 Warning: NaN or Inf found in input tensor

查了查说是因为梯度爆炸或梯度消失，也就是梯度过大或梯度过小。看看模型的参数哪个设置的不太合理（尤其是learning_rate）修改一下再试试。