从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)
答案 0 :(得分:0)
“Momentum”是MomentumOptimizer
分配的插槽的默认名称。
插槽只是一组与训练变量(模型)相关的附加变量(不可训练)。
一旦知道插槽名称,就可以向优化器询问它创建的变量(过滤tf.get_collection()并在var.name中查找“Momentum”):然后可以使用它们来调试训练算法,做关于插槽的统计数据,......
参考文献: