Function step_from_filename

tensorpack/tfutils/varmanip.py:234–237

(name)

        `tf.train.NewCheckpointReader` or :func:`load_checkpoint_vars`.
    """
    def step_from_filename(name):
        name = os.path.basename(name)
        name = name[len(f"{prefix}-"):-len(".index")]
        return int(name)

    checkpoints = glob.glob(os.path.join(dir, "model-*.index"))
    checkpoints = [(f, step_from_filename(f)) for f in checkpoints]