saver.py 文件源码

def get_latest_ckpt(self):
        """Get the latest checkpoint filename in a folder."""

        ckpt_fname_pattern = os.path.join(self.folder, self.fname + '.ckpt*')
        ckpt_fname_list = []
        for fn in os.listdir(self.folder):
            fullname = os.path.join(self.folder, fn)
            if fnmatch.fnmatch(fullname, ckpt_fname_pattern):
                if not fullname.endswith('.meta'):
                    ckpt_fname_list.append(fullname)
        if len(ckpt_fname_list) == 0:
            raise Exception(
                'No checkpoint file found {}'.format(ckpt_fname_pattern))
        ckpt_fname_step = [int(fn.split('-')[-1]) for fn in ckpt_fname_list]
        latest_step = max(ckpt_fname_step)

        latest_ckpt = os.path.join(self.folder,
                                   self.fname + '.ckpt-{}'.format(latest_step))
        latest_graph = os.path.join(self.folder,
                                    self.fname + '.ckpt-{}.meta'.format(latest_step))
        return (latest_ckpt, latest_graph, latest_step)