Update MultKAN.py

KindXiaoming · Jul 15, 2024 · cc9ae32 · cc9ae32
1 parent a47623e
commit cc9ae32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kan/MultKAN.py b/kan/MultKAN.py
@@ -24,7 +24,7 @@
 class MultKAN(nn.Module):
 
     # include mult_ops = []
-    def __init__(self, width=None, grid=3, k=3, mult_arity = 2, noise_scale=1.0, scale_base_mu=0.0, scale_base_sigma=1.0, base_fun='silu', symbolic_enabled=True, affine_trainable=False, grid_eps=1.0, grid_range=[-1, 1], sp_trainable=True, sb_trainable=True, device='cpu', seed=0, save_plot_data=True, sparse_init=False, auto_save=False, first_init=True, ckpt_path='./model', state_id=0):
+    def __init__(self, width=None, grid=3, k=3, mult_arity = 2, noise_scale=1.0, scale_base_mu=0.0, scale_base_sigma=1.0, base_fun='silu', symbolic_enabled=True, affine_trainable=False, grid_eps=1.0, grid_range=[-1, 1], sp_trainable=True, sb_trainable=True, device='cpu', seed=0, save_plot_data=True, sparse_init=False, auto_save=True, first_init=True, ckpt_path='./model', state_id=0):
 
         super(MultKAN, self).__init__()