主要内容

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

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

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

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

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

类别

Baidu
map