GPU提升多分类问题

使用GPU加速多分类问题,与上一篇文章中前几部分代码相同
在这里插入图片描述
GPU提升多分类问题_第1张图片

多层线性网络这里,直接使用nn.Module搭建,与之前自定义搭建的三层网络一样,但注意这里用的激活函数是ReLU的改良后的LeakyReLU,能够避免在输入小于0时出现梯度弥散。
GPU提升多分类问题_第2张图片
设置GPU设备,并把网络和loss函数计算搬到GPU上
在这里插入图片描述
数据也搬到GPU上
GPU提升多分类问题_第3张图片
在任务管理器上看代码运行前后效果,验证是否使用了GPU
GPU提升多分类问题_第4张图片
GPU提升多分类问题_第5张图片

你可能感兴趣的:(分类,数据挖掘,人工智能)