未验证 提交 271d846e 编写于 作者: L LielinJiang 提交者: GitHub

fix first order model output dir error (#62)

上级 100edd84
...@@ -81,6 +81,8 @@ class FirstOrderPredictor(BasePredictor): ...@@ -81,6 +81,8 @@ class FirstOrderPredictor(BasePredictor):
weight_path = get_path_from_url(vox_cpk_weight_url) weight_path = get_path_from_url(vox_cpk_weight_url)
self.weight_path = weight_path self.weight_path = weight_path
if not os.path.exists(output):
os.makedirs(output)
self.output = output self.output = output
self.relative = relative self.relative = relative
self.adapt_scale = adapt_scale self.adapt_scale = adapt_scale
......
...@@ -111,7 +111,7 @@ def _download(url, path, md5sum=None): ...@@ -111,7 +111,7 @@ def _download(url, path, md5sum=None):
raise RuntimeError("Download from {} failed. " raise RuntimeError("Download from {} failed. "
"Retry limit reached".format(url)) "Retry limit reached".format(url))
logger.info("Downloading {} from {}".format(fname, url)) logger.info("Downloading {} from {} to {}".format(fname, url, fullname))
req = requests.get(url, stream=True) req = requests.get(url, stream=True)
if req.status_code != 200: if req.status_code != 200:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册