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.
Categorie
- Built-In Training
Train deep learning networks using built-in training functions
- Custom Training Loops
Train deep learning networks using custom training loops