From de5bf73aa4c1213010499b5d8ea9bd38bcd34f35 Mon Sep 17 00:00:00 2001
From: Mehnaaz <mehnaazmohiuddin@gmail.com>
Date: Tue, 1 Aug 2023 14:00:00 +0530
Subject: [PATCH] Update 04_mnist_basics.ipynb

---
 04_mnist_basics.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/04_mnist_basics.ipynb b/04_mnist_basics.ipynb
index 675bb5b..31436e9 100644
--- a/04_mnist_basics.ipynb
+++ b/04_mnist_basics.ipynb
@@ -4825,10 +4825,10 @@
     "class BasicOptim:\n",
     "    def __init__(self,params,lr): self.params,self.lr = list(params),lr\n",
     "\n",
-    "    def step(self, *args, **kwargs):\n",
+    "    def step(self):\n",
     "        for p in self.params: p.data -= p.grad.data * self.lr\n",
     "\n",
-    "    def zero_grad(self, *args, **kwargs):\n",
+    "    def zero_grad(self):\n",
     "        for p in self.params: p.grad = None"
    ]
   },