guanjihuan/GitHub_collection_pykan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #418 from srigas/master

Browse Source 
`unfix_symbolic_all` produced errors
This commit is contained in:
Ziming Liu 2024-08-27 21:43:14 -04:00 committed by GitHub
commit c0d99819df
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kan/MultKAN.py
View File

@ -951,14 +951,14 @@ class MultKAN(nn.Module):
if log_history:
self.log_history('unfix_symbolic')
def unfix_symbolic_all(self):
def unfix_symbolic_all(self, log_history=True):
'''
unfix all activation functions.
'''
for l in range(len(self.width) - 1):
for i in range(self.width[l]):
for j in range(self.width[l + 1]):
self.unfix_symbolic(l, i, j)
for i in range(self.width_in[l]):
for j in range(self.width_out[l + 1]):
self.unfix_symbolic(l, i, j, log_history)
def get_range(self, l, i, j, verbose=True):
'''