Main Content

本页翻译不是最新的。点击此处可查看最新英文版本。

训练深度神经网络

使用内置训练函数或自定义训练循环来训练网络

定义网络架构后,您可以使用 trainingOptions 函数定义训练参数。然后,您可以使用 trainNetworktrainnet 训练网络。使用经过训练的网络预测类标签或数值响应。

您可以在一个 CPU、一个 GPU、多个 CPU 或 GPU 上训练神经网络,或者在集群中并行训练或在云中训练。在 GPU 上训练或并行训练需要 Parallel Computing Toolbox™。使用 GPU 需要支持的 GPU 设备(有关受支持设备的信息,请参阅GPU Computing Requirements (Parallel Computing Toolbox))。使用 trainingOptions 函数指定执行环境。

如果 trainingOptions 函数不提供任务所需的训练选项,或者自定义输出层不支持所需的损失函数,则您可以定义自定义训练循环。对于无法使用层图创建的网络,可以将自定义网络定义为函数。要了解详细信息,请参阅定义自定义训练循环、损失函数和网络

类别