__init__.py 文件源码

python
阅读 24 收藏 0 点赞 0 评论 0

项目:rnnprop 作者: vfleaking 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号