Skip to content

Commit 985c5eb

Browse files
authored
Update MultKAN.py remove conflict
1 parent 827eab4 commit 985c5eb

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

kan/MultKAN.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -803,13 +803,8 @@ def closure():
803803
train_id = np.random.choice(dataset['train_input'].shape[0], batch_size, replace=False)
804804
test_id = np.random.choice(dataset['test_input'].shape[0], batch_size_test, replace=False)
805805

806-
<<<<<<< HEAD
807806
if _ % grid_update_freq == 0 and _ < stop_grid_update_step and update_grid and _ >= start_grid_update_step:
808807
self.update_grid_from_samples(dataset['train_input'][train_id].to(device))
809-
=======
810-
if _ % grid_update_freq == 0 and _ < stop_grid_update_step and update_grid and _ > start_grid_update_step:
811-
self.update_grid_from_samples(dataset['train_input'][train_id].to(self.device))
812-
>>>>>>> e5abcb74c6cdc6af70d9665ec5a7b4ccb94ee564
813808

814809
if opt == "LBFGS":
815810
optimizer.step(closure)
@@ -1512,4 +1507,4 @@ def get_fun(self, l, i, j):
15121507
plt.plot(inputs, outputs, marker="o")
15131508
return inputs, outputs
15141509

1515-
KAN = MultKAN
1510+
KAN = MultKAN

0 commit comments

Comments
 (0)