diff --git a/reinforcement-learning/mlx-policy-based.ipynb b/reinforcement-learning/mlx-policy-based.ipynb
index 48e970a..5379acb 100644
--- a/reinforcement-learning/mlx-policy-based.ipynb
+++ b/reinforcement-learning/mlx-policy-based.ipynb
@@ -25,7 +25,6 @@
"source": [
"import time\n",
"import imageio\n",
- "import numpy as np\n",
"import mlx.nn as nn\n",
"import mlx.core as mx\n",
"import gymnasium as gym\n",
@@ -232,7 +231,7 @@
" ax[0].axvline(x=i, color='gray', linewidth=0.5)\n",
" ax[1].axvline(x=i, color='gray', linewidth=0.5)\n",
"\n",
- "plt.savefig(\"./static/images/torch-pbg.png\")\n",
+ "plt.savefig(\"./static/images/mlx-pbg.png\")\n",
"plt.show()"
],
"metadata": {
@@ -270,7 +269,7 @@
" state = mx.array(state)\n",
"\n",
" images.append(environment.render())\n",
- "_ = imageio.mimsave('./static/images/torch-pbg.gif', images, duration=25)"
+ "_ = imageio.mimsave('./static/images/mlx-pbg.gif', images, duration=25)"
],
"metadata": {
"collapsed": false,
@@ -284,7 +283,7 @@
{
"cell_type": "markdown",
"source": [
- ""
+ ""
],
"metadata": {
"collapsed": false
diff --git a/reinforcement-learning/mlx-value-based.ipynb b/reinforcement-learning/mlx-value-based.ipynb
index d5e4d3a..d233401 100644
--- a/reinforcement-learning/mlx-value-based.ipynb
+++ b/reinforcement-learning/mlx-value-based.ipynb
@@ -23,10 +23,8 @@
},
"outputs": [],
"source": [
- "import copy\n",
"import time\n",
"import imageio\n",
- "import mlx.nn as nn\n",
"import mlx.core as mx\n",
"import gymnasium as gym\n",
"import mlx.optimizers as optim\n",
@@ -296,7 +294,7 @@
" ax[0].axvline(x=i, color='gray', linewidth=0.5)\n",
" ax[1].axvline(x=i, color='gray', linewidth=0.5)\n",
"\n",
- "plt.savefig(\"./static/images/torch-dqn.png\")\n",
+ "plt.savefig(\"./static/images/mlx-dqn.png\")\n",
"plt.show()"
],
"metadata": {
@@ -333,7 +331,7 @@
" state = mx.array(state)\n",
"\n",
" images.append(environment.render())\n",
- "_ = imageio.mimsave('./static/images/torch-dqn.gif', images, duration=25)"
+ "_ = imageio.mimsave('./static/images/mlx-dqn.gif', images, duration=25)"
],
"metadata": {
"collapsed": false,
@@ -347,7 +345,7 @@
{
"cell_type": "markdown",
"source": [
- ""
+ ""
],
"metadata": {
"collapsed": false