怎么建设网站百度搜索的到在线crm百科

使用GPU加速多分类问题，与上一篇文章中前几部分代码相同

多层线性网络这里，直接使用nn.Module搭建，与之前自定义搭建的三层网络一样，但注意这里用的激活函数是ReLU的改良后的LeakyReLU，能够避免在输入小于0时出现梯度弥散。

设置GPU设备，并把网络和loss函数计算搬到GPU上

数据也搬到GPU上

在任务管理器上看代码运行前后效果，验证是否使用了GPU