使用GPU加速多分类问题,与上一篇文章中前几部分代码相同
多层线性网络这里,直接使用nn.Module搭建,与之前自定义搭建的三层网络一样,但注意这里用的激活函数是ReLU的改良后的LeakyReLU,能够避免在输入小于0时出现梯度弥散。 设置GPU设备,并把网络和loss函数计算搬到GPU上 数据也搬到GPU上 在任务管理器上看代码运行前后效果,验证是否使用了GPU
京公网安备 11010502049817号