为笔记本添加更多功能

Retrieval_based_Voice_Conversion_WebUI.ipynb

@@ -20,11 +20,6 @@
         "!nvidia-smi"
       ]
     },
-    {
-      "cell_type": "markdown",
-      "metadata": {},
-      "source": []
-    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -75,6 +70,7 @@
       },
       "outputs": [],
       "source": [
+        "#@title 下载底模\n",
         "!apt -y install -qq aria2\n",
         "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained/D32k.pth -d /content/Retrieval-based-Voice-Conversion-WebUI/pretrained -o D32k.pth\n",
         "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained/D40k.pth -d /content/Retrieval-based-Voice-Conversion-WebUI/pretrained -o D40k.pth\n",
@@ -106,7 +102,7 @@
         "#@title 从谷歌云盘加载打包好的数据集到/content/dataset\n",
         "\n",
         "#@markdown 数据集位置\n",
-        "DATASET = \"/content/drive/MyDrive/dataset/lulu20230327.zip\"  #@param {type:\"string\"}\n",
+        "DATASET = \"/content/drive/MyDrive/dataset/lulu20230327_32k.zip\"  #@param {type:\"string\"}\n",
         "\n",
         "from google.colab import drive\n",
         "drive.mount('/content/drive')\n",
@@ -141,7 +137,7 @@
         "#@markdown 模型名\n",
         "MODELNAME = \"lulu\"  #@param {type:\"string\"}\n",
         "#@markdown 模型epoch\n",
-        "MODELEPOCH = 3540  #@param {type:\"integer\"}\n",
+        "MODELEPOCH = 6600  #@param {type:\"integer\"}\n",
         "\n",
         "!cp /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/G_{MODELEPOCH}.pth /content/drive/MyDrive/{MODELNAME}_D_{MODELEPOCH}.pth\n",
         "!cp /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/D_{MODELEPOCH}.pth /content/drive/MyDrive/{MODELNAME}_G_{MODELEPOCH}.pth\n",
@@ -165,10 +161,15 @@
         "#@markdown 模型名\n",
         "MODELNAME = \"lulu\"  #@param {type:\"string\"}\n",
         "#@markdown 模型epoch\n",
-        "MODELEPOCH = 730  #@param {type:\"integer\"}\n",
+        "MODELEPOCH = 250  #@param {type:\"integer\"}\n",
+        "\n",
+        "!mkdir -p /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}\n",
         "\n",
         "!cp /content/drive/MyDrive/{MODELNAME}_D_{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/G_{MODELEPOCH}.pth\n",
-        "!cp /content/drive/MyDrive/{MODELNAME}_G_{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/D_{MODELEPOCH}.pth"
+        "!cp /content/drive/MyDrive/{MODELNAME}_G_{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/D_{MODELEPOCH}.pth\n",
+        "!cp /content/drive/MyDrive/*.index /content/\n",
+        "!cp /content/drive/MyDrive/*.npy /content/\n",
+        "!cp /content/drive/MyDrive/{MODELNAME}{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/weights/{MODELNAME}.pth"
       ]
     },
     {
@@ -179,15 +180,107 @@
       },
       "outputs": [],
       "source": [
-        "#@title 手动训练（不推荐）\n",
+        "#@title 手动预处理（不推荐）\n",
         "#@markdown 模型名\n",
         "MODELNAME = \"lulu\"  #@param {type:\"string\"}\n",
         "\n",
-        "!python3 trainset_preprocess_pipeline_print.py /content/dataset 32000 8 logs/{MODELNAME} True\n",
+        "!python3 trainset_preprocess_pipeline_print.py /content/dataset 32000 8 logs/{MODELNAME} True\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "CrxJqzAUKmPJ"
+      },
+      "outputs": [],
+      "source": [
+        "#@title 手动提取特征（不推荐）\n",
+        "#@markdown 模型名\n",
+        "MODELNAME = \"lulu\"  #@param {type:\"string\"}\n",
         "\n",
-        "!python3 extract_feature_print.py 1 0 0 logs/{MODELNAME}\n",
+        "!python3 extract_feature_print.py 1 0 0 logs/{MODELNAME}\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "IMLPLKOaKj58"
+      },
+      "outputs": [],
+      "source": [
+        "#@title 手动训练（不推荐）\n",
+        "#@markdown 模型名\n",
+        "MODELNAME = \"lulu\"  #@param {type:\"string\"}\n",
+        "#@markdown 停止的epoch\n",
+        "MODELEPOCH = 700  #@param {type:\"integer\"}\n",
+        "#@markdown 保存epoch间隔\n",
+        "EPOCHSAVE = 20  #@param {type:\"integer\"}\n",
+        "#@markdown 采样率\n",
+        "MODELSAMPLE = \"48k\"  #@param {type:\"string\"}\n",
         "\n",
-        "!python3 train_nsf_sim_cache_sid_load_pretrain.py -e lulu -sr 32k -f0 0 -g 0 -bs 4 -te 10 -se 5 -pg pretrained/G32k.pth -pd pretrained/D32k.pth -l 0 -c 0\n"
+        "!python3 train_nsf_sim_cache_sid_load_pretrain.py -e lulu -sr {MODELSAMPLE} -f0 1 -bs 32 -g 0 -te {MODELEPOCH} -se {EPOCHSAVE} -pg pretrained/f0G{MODELSAMPLE}.pth -pd pretrained/f0D{MODELSAMPLE}.pth -l 0 -c 1\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "haYA81hySuDl"
+      },
+      "outputs": [],
+      "source": [
+        "#@title 删除其它pth，只留选中的（慎点，仔细看代码）\n",
+        "#@markdown 模型名\n",
+        "MODELNAME = \"lulu\"  #@param {type:\"string\"}\n",
+        "#@markdown 选中模型epoch\n",
+        "MODELEPOCH = 6600  #@param {type:\"integer\"}\n",
+        "\n",
+        "!echo \"备份选中的模型。。。\"\n",
+        "!cp /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/G_{MODELEPOCH}.pth /content/{MODELNAME}_D_{MODELEPOCH}.pth\n",
+        "!cp /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/D_{MODELEPOCH}.pth /content/{MODELNAME}_G_{MODELEPOCH}.pth\n",
+        "\n",
+        "!echo \"正在删除。。。\"\n",
+        "!ls /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}\n",
+        "!rm /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/*.pth\n",
+        "\n",
+        "!echo \"恢复选中的模型。。。\"\n",
+        "!mv /content/{MODELNAME}_D_{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/G_{MODELEPOCH}.pth \n",
+        "!mv /content/{MODELNAME}_G_{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/D_{MODELEPOCH}.pth\n",
+        "\n",
+        "!echo \"删除完成\"\n",
+        "!ls /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "QhSiPTVPoIRh"
+      },
+      "outputs": [],
+      "source": [
+        "#@title 清除项目下所有文件，只留选中的模型（慎点，仔细看代码）\n",
+        "#@markdown 模型名\n",
+        "MODELNAME = \"lulu\"  #@param {type:\"string\"}\n",
+        "#@markdown 选中模型epoch\n",
+        "MODELEPOCH = 1500  #@param {type:\"integer\"}\n",
+        "\n",
+        "!echo \"备份选中的模型。。。\"\n",
+        "!cp /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/G_{MODELEPOCH}.pth /content/{MODELNAME}_D_{MODELEPOCH}.pth\n",
+        "!cp /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/D_{MODELEPOCH}.pth /content/{MODELNAME}_G_{MODELEPOCH}.pth\n",
+        "\n",
+        "!echo \"正在删除。。。\"\n",
+        "!ls /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}\n",
+        "!rm -rf /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/*\n",
+        "\n",
+        "!echo \"恢复选中的模型。。。\"\n",
+        "!mv /content/{MODELNAME}_D_{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/G_{MODELEPOCH}.pth \n",
+        "!mv /content/{MODELNAME}_G_{MODELEPOCH}.pth /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}/D_{MODELEPOCH}.pth\n",
+        "\n",
+        "!echo \"删除完成\"\n",
+        "!ls /content/Retrieval-based-Voice-Conversion-WebUI/logs/{MODELNAME}"
       ]
     }
   ],