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:
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",
")"
]
},