Skip to content

Conversation

@swakhtar
Copy link

Issue: model.train() raises additional argument error because it is calling train() function of torch nn.module, which has only 1 argument.
Fix: fit() method (available in MultKAN class file) is used to train the KAN model.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant