saver.py 文件源码

def __init__(self, folder, fname='model', var_dict=None):

        if not os.path.exists(folder):
            os.makedirs(folder)

        self.folder = folder
        self.log = logger.get()
        self.fname = fname
        self.tf_saver = None
        if var_dict is None:
            self.var_dict = tf.all_variables()
        else:
            self.var_dict = var_dict
        pass