diff --git a/fastNLP/core/callbacks/topk_saver.py b/fastNLP/core/callbacks/topk_saver.py index 0317b7e2..98e5c269 100644 --- a/fastNLP/core/callbacks/topk_saver.py +++ b/fastNLP/core/callbacks/topk_saver.py @@ -36,8 +36,6 @@ class Saver: model_save_fn:Callable=None, **kwargs): if folder is None: folder = Path.cwd().absolute() - if save_object is not None: - logger.info(f"Parameter `folder` is None, and fastNLP will use {folder} to save and load your model.") folder = Path(folder) if not folder.exists(): folder.mkdir(parents=True, exist_ok=True) @@ -209,7 +207,7 @@ class TopkSaver(ResultsMonitor, Saver): if topk is None: topk = 0 ResultsMonitor.__init__(self, monitor, larger_better) - Saver.__init__(self, folder, save_object if topk!=0 else None, only_state_dict, model_save_fn, **kwargs) + Saver.__init__(self, folder, save_object, only_state_dict, model_save_fn, **kwargs) if monitor is not None and topk == 0: raise RuntimeError("`monitor` is set, but `topk` is 0.")