dfd27f0374d3a5ea58a3ae0860ee2649150ac2d0,main.py,,get_latest_model,#,183

Before Change

def get_latest_model():
    train_dirs = glob.glob("%s/*_train" % c.TENSORFLOW_OUT_DIR)
    latest_subdir = max(train_dirs, key=os.path.getmtime)
    if not latest_subdir:
        raise RuntimeError("Can not get latest model, no models found in % s" % c.TENSORFLOW_OUT_DIR)
    latest_model = max(glob.glob("%s/model.ckpt-*.meta" % latest_subdir), key=os.path.getmtime)
    if not latest_model:
        raise RuntimeError("Can not get latest model, no models found in %s" % latest_subdir)
    latest_prefix = latest_model[:-len(".meta")]
    return latest_prefix

After Change

    if not latest_model:
        raise RuntimeError("Can not get latest model, no models found in %s" % c.TENSORFLOW_OUT_DIR)
    latest_prefix = latest_model[:-len(".meta")]
    log.info("Latest model is %s", latest_prefix)
    return latest_prefix


log = logs.get_log(__name__)

Instances