meow!

gistfile1.txt

def get_checkpoints(checkpoint_dir):
    '''
    Finds all checkpoints in a directory and returns them in order
    from least iterations to most iterations
    '''
    meta_list=[]
    for file in os.listdir(checkpoint_dir):
        if file.endswith('.meta'):
            meta_list.append(os.path.join(checkpoint_dir, file[:-5]))
    meta_list = sort_nicely(meta_list)
    return meta_list
    
def sort_nicely(l):
    """
    Sort the given list in the way that humans expect.
    From Ned Batchelder
    https://nedbatchelder.com/blog/200712/human_sorting.html
    """
    def alphanum_key(s):
        """ Turn a string into a list of string and number chunks.
            "z23a" -> ["z", 23, "a"]
        """
        def tryint(s):
            try:
                return int(s)
            except:
                return s
        return [ tryint(c) for c in re.split('([0-9]+)', s) ]
    l.sort(key=alphanum_key)
    return l