Skip to content

Commit c0d9981

Browse files
authored
Merge pull request #418 from srigas/master
`unfix_symbolic_all` produced errors
2 parents 91fc24e + 8a0bdaf commit c0d9981

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

kan/MultKAN.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -951,14 +951,14 @@ def unfix_symbolic(self, l, i, j, log_history=True):
951951
if log_history:
952952
self.log_history('unfix_symbolic')
953953

954-
def unfix_symbolic_all(self):
954+
def unfix_symbolic_all(self, log_history=True):
955955
'''
956956
unfix all activation functions.
957957
'''
958958
for l in range(len(self.width) - 1):
959-
for i in range(self.width[l]):
960-
for j in range(self.width[l + 1]):
961-
self.unfix_symbolic(l, i, j)
959+
for i in range(self.width_in[l]):
960+
for j in range(self.width_out[l + 1]):
961+
self.unfix_symbolic(l, i, j, log_history)
962962

963963
def get_range(self, l, i, j, verbose=True):
964964
'''

0 commit comments

Comments
 (0)