Skip to content

Commit

Permalink
Merge pull request #418 from srigas/master
Browse files Browse the repository at this point in the history
`unfix_symbolic_all` produced errors
  • Loading branch information
KindXiaoming authored Aug 28, 2024
2 parents 91fc24e + 8a0bdaf commit c0d9981
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions kan/MultKAN.py
Original file line number Diff line number Diff line change
Expand Up @@ -951,14 +951,14 @@ def unfix_symbolic(self, l, i, j, log_history=True):
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):
'''
Expand Down

0 comments on commit c0d9981

Please sign in to comment.