pykan
pykan copied to clipboard
cuda support and using KANLayer in pytorch lightning module
Came across an issue when using a KANLayer within a pytorch lightning module and training with cuda. Some tensors being built and processed in the KANLayer were always getting sent to 'cpu'. I resolved this issue and put the updated code on my forked repository, pykan-cuda. Would appreciate if you checked it out.