主要内容

深度学习导入、导出和定制

导入、导出、自定义深度学习网络,自定义层、训练循环、损失函数

从TensorFlow™2、TensorFlow- keras、PyTorch导入网络和层图®ONNX™(开放神经网络交换)模型格式,以及Caffe。您还可以将深度学习工具箱™网络和层图导出到TensorFlow 2和ONNX模型格式。有关更多信息,请参见预训练的深度神经网络

您可以为您的问题定义自己的自定义深度学习层。您可以使用自定义输出层指定自定义损失函数,并定义带有或不带有可学习参数的自定义层。定义自定义层后,可以检查该层是否有效,GPU是否兼容,并输出正确定义的梯度。

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

类别

Baidu
map