We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a29f954 commit f3c857fCopy full SHA for f3c857f
1 file changed
pina/_src/model/block/kan_block.py
@@ -6,7 +6,11 @@
6
from pina._src.model.spline import Spline
7
from pina._src.model.vectorized_spline import VectorizedSpline
8
9
-
+# TODO
10
+# - Improve documentation and comments throughout the code for better clarity.
11
+# - Remove any unused parameters or code related to the base function if it's
12
+# not being utilized in the current implementation.
13
+# - Clean unused code
14
class KANBlock(torch.nn.Module):
15
"""define a KAN layer using splines"""
16
0 commit comments