Siomarry opened this issue 2 years ago · 0 comments
网上说的使用gpu训练有几个地方都需要设置，数据集、损失函数、模型. 代码里面模型的设置. 请问代码可以在gpu环境下运行吗