pykan icon indicating copy to clipboard operation
pykan copied to clipboard

Published 20 hours ago • verified publisher icon KindXiaoming

move weights/biases to target device

Open noskill opened this issue 9 months ago • 1 comments

just add some .to(device) calls

noskill avatar May 03 '24 18:05 noskill

It should fix issue https://github.com/KindXiaoming/pykan/issues/52

noskill avatar May 03 '24 18:05 noskill