pykan
pykan copied to clipboard
How to Use Multi-GPU?
I wanner fit KAN with multi gpus, how to make it by nn.DataParallel(model)?