How to initialise only optimizer variables in Tensorflow?

前端 未结 6 890
别跟我提以往
别跟我提以往 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 05:14

    To fix the None problem just do:

      self.opt_vars = [opt.get_slot(var, name) for name in opt.get_slot_names() 
                       for var in self.vars_to_train
                       if opt.get_slot(var, name) is not None]
    

提交回复
热议问题