AlpsCV

Results 1 comments of AlpsCV

when update the code of https://kindxiaoming.github.io/pykan/API_demo/API_10_device.html from model.train(dataset, opt="LBFGS", steps=50, lamb=5e-5, lamb_entropy=2.); to model.train(dataset, opt="LBFGS", steps=50, lamb=5e-5, lamb_entropy=2., device=device); it will work for both cpu and gpu.