Train Deep Neural Networks
After defining the network architecture, you can define training parameters
using the trainingOptions
function. You can
then train the network using trainNetwork
or trainnet
. Use the trained network to predict class labels or
numeric responses.
You can train a neural network on a CPU, a GPU, multiple CPUs or
GPUs, or in parallel on a cluster or in the cloud. Training on a GPU or in
parallel requires Parallel Computing Toolbox™. Using a GPU requires a supported GPU device (for information on
supported devices, see GPU Computing Requirements (Parallel Computing Toolbox)). Specify
the execution environment using the trainingOptions
function.
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.
Categories
- Built-In Training
Train deep learning networks using built-in training functions
- Custom Training Loops
Train deep learning networks using custom training loops