update 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.")
