base_model.py 文件源码

def __init__(self, env_config, grad_clip_norm=None, log_dir='logs/examples',
                 **kwargs):
        self.env_config = env_config
        self.grad_clip_norm = grad_clip_norm
        self.log_dir = log_dir
        self.placeholders = {}
        self.training_op = None
        self.merged = None
        self._saver = None
        self._writer = None
        self.callbacks = []

        self.global_step_sy = tf.Variable(1, name='global_step', trainable=False)
        placeholders_config = {'add_to_global_step': [[], tf.int32]}
        self._create_placeholders(placeholders_config)
        self.increase_global_step_op = tf.assign_add(
            self.global_step_sy,
            self.placeholders['add_to_global_step'],
            name='increase_global_step')