一、pytorch基础练习
代码中存在的问题:数据类型不匹配
修改之后
相乘需要元素的类型是相同的,修改数据类型之后就可以成功运行了。
其他操作
- 可以进行一些统计方面的工作。
螺旋数据分类
指定设备之后,需要加载模型
可以看到螺旋的边缘数据的分类并不分类
修改之后得到的模型
learning_rate = 1e-3
lambda_l2 = 1e-5
# 这里可以看到,和上面模型不同的是,在两层之间加入了一个 ReLU 激活函数
model = nn.Sequential(
nn.Linear(D, H),
nn.ReLU(),
nn.Linear(H, C)
)
model.to(device)