neural-amp-modeler

commit 8a631bcf96cf74dd823565f3b10ac682b7230b73
parent 413d031b92e011ec0b3e6ab3b865b8632725a219
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Wed, 15 Mar 2023 19:59:12 -0500

Update colab.ipynb (#129)


Diffstat:
M
bin/train/colab.ipynb
 | 
24
++++++++++++++++--------

1 file changed, 16 insertions(+), 8 deletions(-)
diff --git a/bin/train/colab.ipynb b/bin/train/colab.ipynb
@@ -160,15 +160,14 @@
         "model_config = {\n",
         "    \"net\": {\n",
         "        \"name\": \"WaveNet\",\n",
-        "        # This should do decently. If you really want a nice model, try turning up \n",
-        "        # \"channels\" in the first block and \"input_size\" in the second from 12 to 16.\n",
+        "        # This is the \"standard\" model in easy mode / the local GUI trainer.\n",
         "        \"config\": {\n",
         "            \"layers_configs\": [\n",
         "                {\n",
-        "                    \"input_size\": 1,\n",
         "                    \"condition_size\": 1,\n",
+        "                    \"input_size\": 1,\n",
+        "                    \"channels\": 16,\n",
         "                    \"head_size\": 8,\n",
-        "                    \"channels\": 12,\n",
         "                    \"kernel_size\": 3,\n",
         "                    \"dilations\": [1,2,4,8,16,32,64,128,256,512],\n",
         "                    \"activation\": \"Tanh\",\n",
@@ -176,10 +175,10 @@
         "                    \"head_bias\": False\n",
         "                },\n",
         "                {\n",
-        "                    \"input_size\": 12,\n",
         "                    \"condition_size\": 1,\n",
-        "                    \"head_size\": 1,\n",
+        "                    \"input_size\": 16,\n",
         "                    \"channels\": 8,\n",
+        "                    \"head_size\": 1,\n",
         "                    \"kernel_size\": 3,\n",
         "                    \"dilations\": [1,2,4,8,16,32,64,128,256,512],\n",
         "                    \"activation\": \"Tanh\",\n",
@@ -193,7 +192,7 @@
         "        \"val_loss\": \"esr\"\n",
         "    },\n",
         "    \"optimizer\": {\n",
-        "        \"lr\": 0.001\n",
+        "        \"lr\": 0.004\n",
         "    },\n",
         "    \"lr_scheduler\": {\n",
         "        \"class\": \"ExponentialLR\",\n",
@@ -212,7 +211,8 @@
         "    },\n",
         "    \"val_dataloader\": {},\n",
         "    \"trainer\": {\n",
-        "        \"gpus\": 1,\n",
+        "        \"accelerator\": \"gpu\", \n",
+        "        \"devices\": 1,\n",
         "        \"max_epochs\": 100\n",
         "    }\n",
         "}"
@@ -532,7 +532,15 @@
       "name": "python3"
     },
     "language_info": {
+      "codemirror_mode": {
+        "name": "ipython",
+        "version": 3
+      },
+      "file_extension": ".py",
+      "mimetype": "text/x-python",
       "name": "python",
+      "nbconvert_exporter": "python",
+      "pygments_lexer": "ipython3",
       "version": "3.10.6"
     },
     "orig_nbformat": 4,