Remove pretrained folder causing file buildup

BenAAndrew · Oct 5, 2021 · a5a11d1 · a5a11d1
1 parent 1c8385e
commit a5a11d1
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 4 deletions.
diff --git a/application/views.py b/application/views.py
@@ -218,11 +218,9 @@ def train_post():
     metadata_path = os.path.join(paths["datasets"], dataset_name, METADATA_FILE)
     audio_folder = os.path.join(paths["datasets"], dataset_name, AUDIO_FOLDER)
     checkpoint_folder = os.path.join(paths["models"], dataset_name)
-    pretrained_folder = os.path.join(paths["pretrained"], dataset_name)
 
     if request.files.get("pretrained_model"):
-        os.makedirs(pretrained_folder, exist_ok=True)
-        transfer_learning_path = os.path.join(pretrained_folder, "pretrained.pt")
+        transfer_learning_path = os.path.join("data", "pretrained.pt")
         request.files["pretrained_model"].save(transfer_learning_path)
     else:
         transfer_learning_path = None

diff --git a/libdeepspeech.so b/libdeepspeech.so
diff --git a/main.py b/main.py
@@ -13,7 +13,6 @@ def load_paths():
     paths = {
         "datasets": os.path.join("data", "datasets"),
         "models": os.path.join("data", "models"),
-        "pretrained": os.path.join("data", "pretrained"),
         "hifigan": os.path.join("data", "hifigan"),
         "results": os.path.join("data", "results"),
         "languages": os.path.join("data", "languages"),