fix first order model output dir error (PaddlePaddle#62)

AnnaTrainingG · Oct 30, 2020 · 271d846 · 271d846
1 parent 100edd8
commit 271d846
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ppgan/apps/first_order_predictor.py b/ppgan/apps/first_order_predictor.py
@@ -81,6 +81,8 @@ def __init__(self,
                 weight_path = get_path_from_url(vox_cpk_weight_url)
 
         self.weight_path = weight_path
+        if not os.path.exists(output):
+            os.makedirs(output)
         self.output = output
         self.relative = relative
         self.adapt_scale = adapt_scale

diff --git a/ppgan/utils/download.py b/ppgan/utils/download.py
@@ -111,7 +111,7 @@ def _download(url, path, md5sum=None):
             raise RuntimeError("Download from {} failed. "
                                "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)
         if req.status_code != 200: