Main Content

Custom Training Loops

Train deep learning networks using custom training loops

If the trainingOptions function does not provide the training options that you need for your task, or custom output layers do not support the loss functions that you need, then you can define a custom training loop. For networks that cannot be created using layer graphs, you can define custom networks as a function. To learn more, see Define Custom Training Loops, Loss Functions, and Networks.


espandi tutto

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