__init__.py 文件源码

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