深度学习导入、导出和自定义
导入、导出和自定义深度学习网络,并自定义层、训练循环和损失函数
从 TensorFlow™ 2、TensorFlow-Keras、PyTorch®、ONNX™(开放式神经网络交换)模型格式和 Caffe 中导入网络和层图。您还可以将 Deep Learning Toolbox™ 网络和层图导出为 TensorFlow 2 和 ONNX 模型格式。有关详细信息,请参阅 预训练的深度神经网络。
您可以针对您的问题定义自己的自定义深度学习层。您可以使用自定义输出层指定自定义损失函数,并定义具有或不具有可学习参数的自定义层。定义自定义层后,您可以检查该层是否有效,是否与 GPU 兼容,以及是否输出正确定义的梯度。
如果 trainingOptions
函数不提供任务所需的训练选项,或者自定义输出层不支持所需的损失函数,则您可以定义自定义训练循环。对于无法使用层图创建的网络,可以将自定义网络定义为函数。要了解详细信息,请参阅定义自定义训练循环、损失函数和网络。
类别
- 深度学习导入和导出
从外部深度学习平台导入和导出网络
- 深度学习自定义层
为深度学习定义自定义层
- 深度学习自定义训练循环
自定义深度学习训练循环和损失函数