musegan formatting

davidADSP · Apr 15, 2023 · dcfa044 · dcfa044
1 parent 3d28172
commit dcfa044
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/notebooks/11_music/02_musegan/musegan.ipynb b/notebooks/11_music/02_musegan/musegan.ipynb
@@ -139,7 +139,6 @@
     "data_binary = np.eye(N_PITCHES)[two_bars]\n",
     "data_binary[data_binary == 0] = -1\n",
     "data_binary = data_binary.transpose([0, 1, 2, 4, 3])\n",
-    "INPUT_DIM = data_binary.shape[1:]\n",
     "print(f\"Data binary shape {data_binary.shape}\")"
    ]
   },
@@ -325,7 +324,10 @@
    "outputs": [],
    "source": [
     "def Critic():\n",
-    "    critic_input = layers.Input(shape=INPUT_DIM, name=\"critic_input\")\n",
+    "    critic_input = layers.Input(\n",
+    "        shape=(N_BARS, N_STEPS_PER_BAR, N_PITCHES, N_TRACKS),\n",
+    "        name=\"critic_input\",\n",
+    "    )\n",
     "\n",
     "    x = critic_input\n",
     "\n",