diff --git a/fast_bert/learner_util.py b/fast_bert/learner_util.py index 4e0b241..715377a 100644 --- a/fast_bert/learner_util.py +++ b/fast_bert/learner_util.py @@ -128,6 +128,8 @@ def save_model(self, path=None): if not path: path = self.output_dir / "model_out" + else: + path = Path(path) path.mkdir(exist_ok=True)