2.1.4 定义损失函数和优化器

使用分类交叉熵（CrossEntropy）作为损失函数，支持动量的SGD作为优化器，代码如下：

这里设置动量为0.9，这也是深度学习中最常用的参数。