Remove parameters from discriminator optimizer update

patrick-kidger · Feb 4, 2025 · 550967c · 550967c
1 parent f8df0e6
commit 550967c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/examples/deep_convolutional_gan.ipynb b/examples/deep_convolutional_gan.ipynb
@@ -431,9 +431,7 @@
     "        key,\n",
     "    )\n",
     "\n",
-    "    updates, opt_state = discriminator_optimizer.update(\n",
-    "        grads, discriminator_opt_state, discriminator\n",
-    "    )\n",
+    "    updates, opt_state = discriminator_optimizer.update(grads, discriminator_opt_state)\n",
     "    discriminator = eqx.apply_updates(discriminator, updates)\n",
     "\n",
     "    return loss, discriminator, opt_state, generator_state, discriminator_state, key\n",