def __init__(self, name, optimizee=None, n_bptt_steps=None, lr=1e-4, use_avg_loss=False, is_training=True, optimizer_name='adam', **kwargs):
self.name = name
self.is_training = is_training
self.kwargs = kwargs
if self.is_training:
self.optimizee = optimizee
self.optimizer_name = optimizer_name
self.x_dim = optimizee.get_x_dim()
self.f = optimizee.loss
self.n_bptt_steps = n_bptt_steps
self.train_lr = lr
self.use_avg_loss = use_avg_loss
else:
self.x_dim = 233
self.session = tf.get_default_session()
self._build()
self.bid = 0
评论列表
文章目录