Skip to content

GitLab

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

fix ckp loading logging

parent 486f4f3d
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
unicore/trainer.py
View file @ f24a5f70
...@@ -366,6 +366,7 @@ class Trainer(object): ...@@ -366,6 +366,7 @@ class Trainer(object):
dist_device=self.device, dist_device=self.device,
) )
had_loaded_model = False
if bexists: if bexists:
state = None state = None
if is_master: if is_master:
...@@ -396,6 +397,7 @@ class Trainer(object): ...@@ -396,6 +397,7 @@ class Trainer(object):
) )
# save memory for later steps # save memory for later steps
del state["model"] del state["model"]
had_loaded_model = True
if errors.missing_keys: if errors.missing_keys:
logger.warning( logger.warning(
...@@ -485,10 +487,10 @@ class Trainer(object): ...@@ -485,10 +487,10 @@ class Trainer(object):
) )
) )
elif had_loaded_model:
logger.info("Loaded checkpoint {}".format(filename))
else: else:
logger.info( logger.info("No existing checkpoint found {}".format(filename))
"Loaded checkpoint {}".format(filename)
)
return extra_state 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