-
Notifications
You must be signed in to change notification settings - Fork 9
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Adding other KAN layers #3
Conversation
Thanks @armbrusl, the code looks ok. Can you
|
just pushed a commit to master with all the Lux 0.x --> 1.x fixes. @armbrusl I also rebased this branch to master. I removed all the unnecessary packages in Can you also move the test you added in |
I think I have addressed your suggestions. Thanks for walking me through it! Let me know if there is anything else I need to do. The only thing I haven't done is edit the README file, is there anything specific I should add ? |
see above comments |
# Batched Multiplication using Einstein Summation | ||
#======================================================# | ||
|
||
@inline function batched_mul(x, coeffs) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you not use NNlib.batched_mul or NNlib.batched_vec for this?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I am not sure. I have been trying but I have not able to get it right
Can you add the new layers to the code in the README and report timings? https://github.com/vpuri3/KolmogorovArnold.jl/blob/master/README.md?plain=1#L14-L88 If not, I can do it after this is merged. |
@armbrusl can you investigate this error? I am not seeing it in master with 1.10 or 1.11. I also updated master to only test on 1.10 and 1.11. If you rebase this branch, then that change will be incorporated. |
@vpuri3 I am not able to replicate the errors, when I run the required command import Pkg; Pkg.test(;coverage=true, julia_args=["--check-bounds=yes", "--compiled-modules=yes"], force_latest_compatible_version=false, allow_reresolve=true) everything works fine. I also rebased the branch. |
@armbrusl i tested the CI on Master in #4 and it is fine. The failures here are due to changes introduced in this pr I also don't see your rebase commits. Make sure you git checkout master, git fetch and git pull the latest version. then checkout this branch again and git merge master. and then do a git push |
I think I have done the steps you told me, how will I know if I have done it right |
I'll check out this branch and investigate this evening |
@armbrusl you haven't rebased correctly. Look at this branch. This branch is some commits behind. You need to rebase this branch to the latest master. |
I have implemented the following two layers
Julia implementation of FourierKAN
Julia implementation of ChebyKAN
I also added another test for all the layers. Not sure how most of GitHub works so let me know if there is anything else I should do.