pykan icon indicating copy to clipboard operation
pykan copied to clipboard

cuda support and using KANLayer in pytorch lightning module

Open ethanpirso opened this issue 9 months ago • 0 comments

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.

ethanpirso avatar May 14 '24 16:05 ethanpirso