def saver(self, **kwargs):
"""Returns a Saver for all (trainable and model) variables used by the model.
Model variables include e.g. moving mean and average in BatchNorm.
:return: tf.Saver
"""
return tf.train.Saver(self.vars, **kwargs)
评论列表
文章目录