need long time to init efficient-KAN

[TuuSiwei]

When I initialize efficient KAN with a high number of nodes, I have to wait a long time

my test code is as follows：

from kan import KAN
import time

start_time=time.time()
model=KAN([1024,3584,3584])
print(f'Initialization time: {time.time()-start_time}')

and the output is：

Initialization time: 330.3829219341278

Why does it take so long to initialize

[TuuSiwei]

After a single step, I found that the problem came from the curve2coeff function's torch.linalg.lstsq(), which was taking an enormous amount of time

[lsz92cyn]

Do you have any ways to solve the issue of slow speed?

[TuuSiwei]

Not yet, I only found this problem on Ubuntu servers, not on windows laptops. Maybe it has something to do with cpu performance