Commit f24a5f70 authored by Guolin Ke's avatar Guolin Ke
Browse files

fix ckp loading logging

parent 486f4f3d
......@@ -366,6 +366,7 @@ class Trainer(object):
dist_device=self.device,
)
had_loaded_model = False
if bexists:
state = None
if is_master:
......@@ -396,6 +397,7 @@ class Trainer(object):
)
# save memory for later steps
del state["model"]
had_loaded_model = True
if errors.missing_keys:
logger.warning(
......@@ -485,10 +487,10 @@ class Trainer(object):
)
)
elif had_loaded_model:
logger.info("Loaded checkpoint {}".format(filename))
else:
logger.info(
"Loaded checkpoint {}".format(filename)
)
logger.info("No existing checkpoint found {}".format(filename))
return extra_state
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment