可以改为使用GPU进行训练么

[luckyooh]

如题

[A-biao96]

可以，加一行代码。
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

不过会有警告，起不到加速作用，并没有用cpu运行的快，原因和代码编写有关，目前还没解决。

[zepen]

这点数据量和模型参数，GPU犹如杀鸡用牛刀