neural-amp-modeler

commit f7960499a060ad7f725634c1c331a01548a0a733
parent 6f9b7245f3b7540b8472cd4454d4ff55d4977604
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Sat, 13 Apr 2024 11:06:51 -0700

[ENHANCEMENT] Put back manual reamp latency compensation in Colab (#398)

Add back visible latency setting option to easy_colab.ipynb
Diffstat:
M
bin/train/easy_colab.ipynb
 | 
21
+++++++++++++++++----

1 file changed, 17 insertions(+), 4 deletions(-)
diff --git a/bin/train/easy_colab.ipynb b/bin/train/easy_colab.ipynb
@@ -36,7 +36,7 @@
         "## Step 2: Train!\n",
         "Configure your training run below, then hit the Play button to start training!\n",
         "\n",
-        "🕙NOTE: At default settings, training will take just over 10 minutes.🕙"
+        "🕙NOTE: At default settings, training will take about 10 minutes.🕙"
       ]
     },
     {
@@ -69,6 +69,7 @@
         "#@markdown # Training parameters\n",
         "epochs = 100 #@param {type: \"number\"}\n",
         "architecture = \"standard\"  #@param [\"standard\", \"lite\", \"feather\", \"nano\"] {type: \"string\"}\n",
+        "latency_samples = \"0\"  #@param {type: \"string\"}\n",
         "fit_cab = False  #@param {type: \"boolean\"}\n",
         "ignore_checks = False #@param {type: \"boolean\"}\n",
         "\n",
@@ -78,8 +79,8 @@
         "modeled_by = \"Your name\" #@param {type:\"string\"}\n",
         "gear_make = \"GearCo\" #@param {type:\"string\"}\n",
         "gear_model = \"GearName\" #@param {type:\"string\"}\n",
-        "gear_type = \"pedal\" #@param [\"amp\", \"pedal\", \"pedal_amp\", \"amp_cab\", \"amp_pedal_cab\", \"preamp\", \"studio\"] {type:\"string\"}\n",
-        "tone_type = \"overdrive\" #@param [\"clean\", \"overdrive\", \"crunch\", \"hi_gain\", \"fuzz\"] {type:\"string\"}\n",
+        "gear_type = \"amp\" #@param [\"amp\", \"pedal\", \"pedal_amp\", \"amp_cab\", \"amp_pedal_cab\", \"preamp\", \"studio\"] {type:\"string\"}\n",
+        "tone_type = \"clean\" #@param [\"clean\", \"overdrive\", \"crunch\", \"hi_gain\", \"fuzz\"] {type:\"string\"}\n",
         "\n",
         "def _verbose_enum(E, val):\n",
         "    try:\n",
@@ -91,6 +92,17 @@
         "            + \", \".join(list(x.value for x in E))\n",
         "        )\n",
         "\n",
+        "def _parse_latency(ls: str):\n",
+        "    if ls.lower() == \"auto\":\n",
+        "        return None\n",
+        "    try:\n",
+        "        return int(ls)\n",
+        "    except ValueError as e:\n",
+        "        raise ValueError(\n",
+        "            f\"Invalid value for latency {ls} was given. Either use 'auto' or provide \"\n",
+        "            f\"the reamp latency, in samples.\\nOriginal error:\\n\\n{e}\"\n",
+        "        )\n",
+        "\n",
         "user_metadata = None if not use_metadata else UserMetadata(\n",
         "    name=name,\n",
         "    modeled_by=modeled_by,\n",
@@ -107,7 +119,8 @@
         "    architecture=architecture,\n",
         "    fit_cab=fit_cab,  # Change me to True for full-rig modeling!\n",
         "    ignore_checks=ignore_checks,  # Change to True to train anyways with potentially bad data\n",
-        "    user_metadata=user_metadata\n",
+        "    user_metadata=user_metadata,\n",
+        "    delay=_parse_latency(latency_samples)\n",
         ")"
       ]
     },