Main Content

自定义训练循环

使用自定义训练循环来训练深度学习网络

如果 trainingOptions 函数不提供任务所需的训练选项,或您有 trainnet 函数不支持的损失函数,您可以定义自定义训练循环。对于无法指定为由层组成的网络的模型,可以将模型定义为函数。要了解详细信息,请参阅定义自定义训练循环、损失函数和网络

函数

全部展开

dlnetworkDeep learning neural network (自 R2019b 起)
trainingProgressMonitorMonitor and plot training progress for deep learning custom training loops (自 R2022b 起)
minibatchqueueCreate mini-batches for deep learning (自 R2020b 起)
padsequencesPad or truncate sequence data to same length (自 R2021a 起)
dlarrayDeep learning array for customization (自 R2019b 起)
dlgradientCompute gradients for custom training loops using automatic differentiation (自 R2019b 起)
dlfevalEvaluate deep learning model for custom training loops (自 R2019b 起)
crossentropyCross-entropy loss for classification tasks (自 R2019b 起)
l1lossL1 loss for regression tasks (自 R2021b 起)
l2lossL2 loss for regression tasks (自 R2021b 起)
huberHuber loss for regression tasks (自 R2021a 起)
mseHalf mean squared error (自 R2019b 起)
ctcConnectionist temporal classification (CTC) loss for unaligned sequence classification (自 R2021a 起)

主题