KAN.train doesn't allow to use KAN inside other torch models

[noskill]

KAN implementation overrides https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module.train If i embed KAN in a pytorch module and call a train on it it causes error, because KAN's train expects a dict

[fuhongxue00]

This does hinder us form using KAN for testing. I just tried to embed KAN in my model but have no idea how to code