Merge pull request #228 from Kayzwer/master

fix error in docs
KindXiaoming · May 23, 2024 · 27b4640 · 27b4640
2 parents d49d677 + a3ac5d6
commit 27b4640
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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