fixed

kan-bayashi · Aug 18, 2020 · 1ca8781 · 1ca8781
1 parent 93dfc81
commit 1ca8781
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parallel_wavegan/utils/utils.py b/parallel_wavegan/utils/utils.py
@@ -337,6 +337,6 @@ def download_pretrained_model(tag, download_dir=None):
         gdown.download(f"https://drive.google.com/uc?id={id_}", output_path, quiet=False)
         with tarfile.open(output_path, 'r:*') as tar:
             tar.extractall(f"{download_dir}/{tag}")
-    checkpoint_path = find_files(f"{download_dir}/{tag}", "*.pkl")
+    checkpoint_path = find_files(f"{download_dir}/{tag}", "checkpoint*.pkl")
 
     return checkpoint_path[0]