使用MindSpore优化模型参数

在训练神经网络模型过程中，必不可少的是需要定义损失函数和优化器。下面介绍下使用mindspore深度学习框架如何调用。

1. MindSpore支持的损失函数有SoftmaxCrossEntropyWithLogits、L1Loss、MSELoss等。这里使用交叉熵损失函数SoftmaxCrossEntropyWithLogits。

   # 定义损失函数

   net_loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction='mean')

2. MindSpore支持的优化器有Adam、AdamWeightDecay、Momentum等。这里使用Momentum优化器为例。

   # 定义优化器

   net_opt = nn.Momentum(net.trainable_params(), learning_rate=0.01, momentum=0.9)