diff --git a/utils.py b/utils.py index 36caec2..a5dae7b 100644 --- a/utils.py +++ b/utils.py @@ -36,8 +36,7 @@ def load_checkpoint(checkpoint_path, model, optimizer=None, skip_optimizer=False new_state_dict = {} for k, v in state_dict.items(): try: - #assert "emb_g" not in k - # print("load", k) + # assert "emb_g" not in k new_state_dict[k] = saved_state_dict[k] assert saved_state_dict[k].shape == v.shape, (saved_state_dict[k].shape, v.shape) except: @@ -80,7 +79,6 @@ def latest_checkpoint_path(dir_path, regex="G_*.pth"): f_list = glob.glob(os.path.join(dir_path, regex)) f_list.sort(key=lambda f: int("".join(filter(str.isdigit, f)))) x = f_list[-1] - print(x) return x