How to initialise only optimizer variables in Tensorflow?

前端 未结 6 888
别跟我提以往
别跟我提以往 2020-12-13 04:27

I want to use MomentumOptimizer in Tensorflow. However, since this optimizer uses some internal variable, attempting to use it without initializing this variabl

6条回答
  •  鱼传尺愫
    2020-12-13 04:58

    You can filter variables by name and only initialize those. IE

    momentum_initializers = [var.initializer for var in tf.global_variables() if 'Momentum' in var.name]
    sess.run(momentum_initializers)
    

提交回复
热议问题