How do I know and tune the batch size while using patternnet for training a neural network classification model
24 vues (au cours des 30 derniers jours)
Afficher commentaires plus anciens
So I tried using the patternet to classify the type of wines using the inbult wines dataset. I was wondering, if there was an option to tune the parameter such as batch size and etc? I know how to change the training fucntion and cost function but I'm unable to figure out how to tweak or tune the batch size. Any help would be appreciated.
0 commentaires
Réponses (1)
Sahil Jain
le 19 Oct 2021
Hi. As of 2021b, shallow networks such as "patternnet" do not support minibatch training. The training function "traingd" uses the batch gradient descent algorithm and uses the entire dataset for each epoch. For minibatch training, consider using a deep neural network with the "trainNetwork" function. The Create Simple Image Classification Network documentation page will help you get started.
0 commentaires
Voir également
Catégories
En savoir plus sur Deep Learning Toolbox dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!