fix colab

blaisewf · Apr 14, 2024 · b40fa08 · b40fa08
1 parent 0cde185
commit b40fa08
Showing 1 changed file with 21 additions and 19 deletions.
diff --git a/RVC_CLI.ipynb b/RVC_CLI.ipynb
@@ -431,6 +431,8 @@
         "save_every_epoch = 10  # @param {type:\"slider\", min:1, max:100, step:0}\n",
         "save_only_latest = False  # @param{type:\"boolean\"}\n",
         "save_every_weights = False  # @param{type:\"boolean\"}\n",
+        "overtraining_detector = False  # @param{type:\"boolean\"}\n",
+        "overtraining_threshold = 50  # @param {type:\"slider\", min:1, max:100, step:0}\n",
         "# @markdown ### ❓ Optional\n",
         "# @markdown In case you select custom pretrained, you will have to download the pretraineds and enter the path of the pretraineds.\n",
         "custom_pretrained = False  # @param{type:\"boolean\"}\n",
@@ -454,7 +456,7 @@
         "    if tensorboard == True:\n",
         "        %load_ext tensorboard\n",
         "        %tensorboard --logdir /content/RVC_CLI/logs/\n",
-        "    !python main.py train --model_name \"{model_name}\" --rvc_version \"{rvc_version}\" --save_every_epoch \"{save_every_epoch}\" --save_only_latest \"{save_only_latest}\" --save_every_weights \"{save_every_weights}\" --total_epoch \"{total_epoch}\" --sampling_rate \"{sr}\" --batch_size \"{batch_size}\" --gpu \"{gpu}\" --pitch_guidance \"{pitch_guidance}\" --pretrained \"{pretrained}\" --custom_pretrained \"{custom_pretrained}\" --g_pretrained_path \"{g_pretrained_path}\" --d_pretrained_path \"{d_pretrained_path}\"\n",
+        "    !python main.py train --model_name \"{model_name}\" --rvc_version \"{rvc_version}\" --save_every_epoch \"{save_every_epoch}\" --save_only_latest \"{save_only_latest}\" --save_every_weights \"{save_every_weights}\" --total_epoch \"{total_epoch}\" --sampling_rate \"{sr}\" --batch_size \"{batch_size}\" --gpu \"{gpu}\" --pitch_guidance \"{pitch_guidance}\" --pretrained \"{pretrained}\" --custom_pretrained \"{custom_pretrained}\" --g_pretrained_path \"{g_pretrained_path}\" --d_pretrained_path \"{d_pretrained_path}\" --overtraining_detector \"{overtraining_detector}\" --overtraining_threshold \"{overtraining_threshold}\"\n",
         "\n",
         "\n",
         "server_thread = threading.Thread(target=start_train)\n",
@@ -544,15 +546,21 @@
     },
     {
       "cell_type": "markdown",
-      "source": [
-        "# Resume-training"
-      ],
       "metadata": {
         "id": "OaKoymXsyEYN"
-      }
+      },
+      "source": [
+        "# Resume-training"
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "cellView": "form",
+        "id": "d3KgLAYnyHkP"
+      },
+      "outputs": [],
       "source": [
         "# @title Load a Backup\n",
         "from google.colab import drive\n",
@@ -573,16 +581,16 @@
         "        shutil.copy(time_, time__)\n",
         "    shutil.copytree(source_path, destination_path)\n",
         "    print(\"Model backup loaded successfully.\")"
-      ],
-      "metadata": {
-        "cellView": "form",
-        "id": "d3KgLAYnyHkP"
-      },
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "cellView": "form",
+        "id": "sc9DzvRCyJ2d"
+      },
+      "outputs": [],
       "source": [
         "# @title Set training variables\n",
         "# @markdown ### ➡️ Use the same as you did previously\n",
@@ -592,13 +600,7 @@
         "f0method = \"rmvpe\"  # @param [\"pm\", \"dio\", \"crepe\", \"crepe-tiny\", \"harvest\", \"rmvpe\"] {allow-input: false}\n",
         "hop_length = 128  # @param {type:\"slider\", min:1, max:512, step:0}\n",
         "sr = int(sample_rate.rstrip(\"k\")) * 1000"
-      ],
-      "metadata": {
-        "cellView": "form",
-        "id": "sc9DzvRCyJ2d"
-      },
-      "execution_count": null,
-      "outputs": []
+      ]
     }
   ],
   "metadata": {