Momentum在更快的RCNN变量的名称或变量范围中意味着什么?

时间:2017-10-05 05:21:18

标签: python python-2.7 tensorflow object-detection tensorflow-gpu

从tf.get_collection()

打印下面的行
Tensor("conv3_1/weights/Momentum/read:0", shape=(3, 3, 128, 256), dtype=float32)
Tensor("conv3_1/biases/Momentum/read:0", shape=(256,), dtype=float32)
Tensor("conv3_2/weights/Momentum/read:0", shape=(3, 3, 256, 256), dtype=float32)
Tensor("conv3_2/biases/Momentum/read:0", shape=(256,), dtype=float32)

1 个答案:

答案 0 :(得分:0)

“Momentum”是MomentumOptimizer分配的插槽的默认名称。

插槽只是一组与训练变量(模型)相关的附加变量(不可训练)。

一旦知道插槽名称,就可以向优化器询问它创建的变量(过滤tf.get_collection()并在var.name中查找“Momentum”):然后可以使用它们来调试训练算法,做关于插槽的统计数据,......

参考文献:

广告位说明:https://github.com/tensorflow/tensorflow/blob/97af82d5327e2e0142d8c572279b5282b7b17d92/tensorflow/python/training/optimizer.py#L235

名称“动量”:https://github.com/tensorflow/tensorflow/blob/97af82d5327e2e0142d8c572279b5282b7b17d92/tensorflow/python/training/momentum.py#L52