tensorflow2.0 怎么使用L-BFGS优化方法?

如题。if.contrib.opt.ScipyOptimizerInterface被弃用了,替换其他优化器好像又不太行。
具体来说就是如下代码怎么在tf2.0使用啊。

self.optimizer = tfp.contrib.opt.ScipyOptimizerInterface(self.loss,
                                                                method = 'L-BFGS-B',
                                                                options = {'maxiter': 50000,
                                                                           'maxfun': 50000,
                                                                           'maxcor': 50,
                                                                           'maxls': 50,
                                                                           'ftol' : 1.0 * np.finfo(float).eps})

                                                                                                                                                            self.optimizer.minimize(self.sess,
                                feed_dict = tf_dict,
                                fetches = [self.loss],
                                loss_callback = self.callback)
查看全部
qq_42105362
qq_42105362
2020/08/06 10:19
  • tensorflow
  • 神经网络
  • 点赞
  • 收藏
  • 回答
    私信

4个回复