From a3ac5d61135a48cb8de7a0cd13a609f54851213a Mon Sep 17 00:00:00 2001
From: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
Date: Wed, 22 May 2024 12:30:05 +0800
Subject: [PATCH] fix error in docs

---
 kan/KAN.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kan/KAN.py b/kan/KAN.py
index 28b0aa26..4e981d0e 100644
--- a/kan/KAN.py
+++ b/kan/KAN.py
@@ -25,7 +25,7 @@ class KAN(nn.Module):
         depth: int
             depth of KAN
         width: list
-            number of neurons in each layer. e.g., [2,5,5,3] means 2D inputs, 5D outputs, with 2 layers of 5 hidden neurons.
+            number of neurons in each layer. e.g., [2,5,5,3] means 2D inputs, 3D outputs, with 2 layers of 5 hidden neurons.
         grid: int
             the number of grid intervals
         k: int