提交 301cdc81 编辑于 作者: Toshihiro Nakae's avatar Toshihiro Nakae
浏览文件

check existance of model directory in restore

上级 2143823e

dagmm/dagmm.py

+2 −0
原始行号 差异行号 差异行
@@ -190,6 +190,8 @@ class DAGMM: 
        fdir : str

            Path of directory trained model is saved.

        """

        if not exists(fdir):

            raise Exception("Model directory does not exist.")



        model_path = join(fdir, self.MODEL_FILENAME)

        meta_path = model_path + ".meta"