How to embed KAN into other models?

[mhy9989]

I used model = KAN(layers,3,3).speed() to create a KAN model and embedded it into another model to replace MLP. However, I found that the training speed was very slow during the training process. How can I use KAN as a submodule correctly?

[limllzu]

我使用 model = KAN(layers,3,3).speed() 创建了一个 KAN 模型，并将其嵌入到另一个模型中以替代 MLP。但是在训练过程中，我发现训练速度非常慢。如何正确使用 KAN 作为子模块？

你好，请问你是怎么解决这个问题的，我现在遇到的问题也是训练过程中推理速度特别慢，谢谢

[mhy9989]

I am currently using the efficient-kan, you can try the model in this repository: https://github.com/Blealtan/efficient-kan