mindspore.nn提供了常见神经网络层的实现,可以使用基本单元Cell来自定义模型类,而通过一个nn.Softmax
层实例,可以获得预测概率。模型层可以使用多种方法构建,多尝试不同的模型层,可以找到适合的方式。模型参数可以通过 model.parameters_and_names()
来获取。
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交