【问题标题】:After tensorflow training, vgg-net return nantensorflow训练后,vgg-net返回nan
【发布时间】:2018-10-30 21:27:07
【问题描述】:

我训练 vgg-19 net 到 classfy cifar10,训练一次后,vgg-net 返回nan

0 [[  4.45161677e+09   2.87961518e+10   4.20765041e+10 ...,          -2.33432433e+10
1.83500431e+10  -1.12923648e+10]
 [  1.18354002e+10   3.38799473e+10   5.86873242e+10 ...,  -4.18343895e+10
2.79392338e+10  -1.61746637e+10]
 [  1.26074880e+09   2.22301839e+10   5.25488333e+10 ...,  -2.92738212e+10
2.51925299e+10  -1.48290714e+10]
 ..., 
 [  1.05694116e+10   2.16351908e+10   5.02961357e+10 ...,  -3.12492278e+10
2.42959094e+10  -1.26112993e+10]
 [  4.72429568e+09   2.75032003e+10   5.14044682e+10 ...,  -3.51395635e+10
2.18048840e+10  -1.46147287e+10]
 [  2.97774285e+09   1.89559747e+10   4.06387917e+10 ...,  -2.35828470e+10
1.96148122e+10  -9.55916698e+09]]
1 [[ nan  nan  nan ...,  nan  nan  nan]
 [ nan  nan  nan ...,  nan  nan  nan]
 [ nan  nan  nan ...,  nan  nan  nan]
 ..., 
 [ nan  nan  nan ...,  nan  nan  nan]
 [ nan  nan  nan ...,  nan  nan  nan]
 [ nan  nan  nan ...,  nan  nan  nan]]

我使用tf.train.GradientDescentOptimizer训练vgg net,活动函数是relu,tf.random_normal初始化权重并使用tf.nn.xw_plus_b作为全连接层。所以我想知道,为什么 vgg-net 在训练后返回nan

【问题讨论】:

  • 问题解决了。在网络中,我的学习率太大(1e-3),我将 lr 更改为 1e-11,开始训练。非常感谢。

标签: python tensorflow deep-learning vgg-net


【解决方案1】:

降低学习率可以解决这个数值稳定性问题。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2016-11-16
    • 2017-07-08
    • 2019-12-20
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多