CPU以外対応 - Githubissues

o-jill / tigerdenversi

machine learning for ruversi.

0 stars 0 forks source link

Closed o-jill closed 1 week ago

o-jill commented 3 weeks ago

CPU以外のデバイス(CUDA, MPS)で動くようにする。起動時オプションで使いたいデバイスを指定できるようにする。

o-jill commented 3 weeks ago

MPS指定でMPSが使えないときは遠慮なくpanic!()しちゃおうか。

o-jill commented 1 week ago

to_device()するタイミング？とりあえずshuffle()してからやってるけど、もしかしたら予めinputとtargetをそれぞれto_device()出来るのではないか？epochごとのto_device()が無くせるのではないか？