Skip to content

Commit

Permalink
🗑 Cleaned up.
Browse files Browse the repository at this point in the history
  • Loading branch information
BrikerMan committed Oct 28, 2020
1 parent 990be62 commit 345cf80
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions kashgari/tasks/abs_task_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit 345cf80

Please sign in to comment.