Cyrus Pellet
Cyrus Pellet
Kindly merge this please
It turns out that replacing `model = KAN(width=[784,5,5,10], grid=3, k=3).to(device)` by `model = KAN(width=[784,5,5,10], grid=3, k=3, device=device)` does the trick for me! Here is a full example training on mps...
No unfortunately, @WuZhuoran's comments make sense, I was solely making a point about getting training on mps to work
You can use $\text{}$
In this case allow me to suggest to make this clearer; going out of scope is not the issue here, interacting with the object in a single clause will not...