From 345cf808eeb3465af774413c021d9861f4d2ec58 Mon Sep 17 00:00:00 2001 From: BrikerMan Date: Wed, 28 Oct 2020 22:01:09 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=97=91=20Cleaned=20up.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- kashgari/tasks/abs_task_model.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/kashgari/tasks/abs_task_model.py b/kashgari/tasks/abs_task_model.py index b40c9bc4..6da8ea99 100644 --- a/kashgari/tasks/abs_task_model.py +++ b/kashgari/tasks/abs_task_model.py @@ -76,13 +76,7 @@ def default_hyper_parameters(cls) -> Dict[str, Dict[str, Any]]: """ raise NotImplementedError - def save(self, model_path: str, h5_weight: bool = False) -> str: - """ - Save model - Args: - model_path: target model path - h5_weight: whether using original h5 format or new saved_model format - """ + def save(self, model_path: str) -> str: pathlib.Path(model_path).mkdir(exist_ok=True, parents=True) model_path = os.path.abspath(model_path)