Skip to content
代码片段 群组 项目
提交 301cdc81 编辑于 作者: Toshihiro Nakae's avatar Toshihiro Nakae
浏览文件

check existance of model directory in restore

上级 2143823e
无相关合并请求
......@@ -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"
......
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册