Borrar filtros
Borrar filtros

How to see the training progress in CNN?

10 visualizaciones (últimos 30 días)
Stu
Stu el 27 de Mzo. de 2018
Comentada: Mibang el 22 de Dic. de 2023
When the CNN training is running, how to make it show the training progress (the table with iteration no., training accuracy, loss etc.). I asked for the second output through "info" in the trainNetwork function which gave me a summary at the end. But I want to see it while the training is going on.

Respuesta aceptada

Javier Pinzón
Javier Pinzón el 6 de Abr. de 2018
Editada: Javier Pinzón el 6 de Abr. de 2018
Hello Stu,
In the trainingOptions, you need tu put the option 'Verbose', true, ... , in that way you can see the progress, and set your VerboseFrequency, also, if you are using the 2017b, you can also plot your progress, using the option 'Plots','training-progress', ...
So, a full example may be:
opts = trainingOptions('sgdm', ...
'InitialLearnRate', 0.01, ...
'L2Regularization', 0.004, ...
'Shuffle', 'every-epoch', ...
'MaxEpochs', 20, ...
'MiniBatchSize', 69, ...
'Verbose', true, ...
'VerboseFrequency', 320, ...
'ValidationData',ValidationDS, ...
'ValidationFrequency',160, ...
'ValidationPatience',Inf, ...
'Plots','training-progress', ...
'ExecutionEnvironment', 'auto', ...
'CheckpointPath','data\epoch');
Also I recommend you yo use a MiniBatchSize multiple of your total amount of images of your training dataset, and the VerboseFrequency, be set as TotalImage/MiniBatchSize in that way you will see what happens at the end of every epoch, as in the example, there are a total of 22080 images, and I set 69 as the batch size, so, to see it in every epoch, i said that i wanted to see every 320 iterations the table (320 iter = 1 Epoch).
Hope it may help you.
Regards, Javier
  1 comentario
Mibang
Mibang el 22 de Dic. de 2023
Great, thanks, I also took your answer for my work.

Iniciar sesión para comentar.

Más respuestas (0)

Categorías

Más información sobre GPU Computing en Help Center y File Exchange.

Etiquetas

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by