Effacer les filtres
Effacer les filtres

[Need Help] Cannot use DL checkpoint model to predict?

2 vues (au cours des 30 derniers jours)
Song Decn
Song Decn le 27 Mar 2020
Commenté : Song Decn le 8 Fév 2021
I tried to use models saved in path specified by 'CheckpointPath' parameter during training by Deep Learning Toolbox, I got following error:
Error using DAGNetwork/calculatePredict>predictSingle (line 112)
Input parameter has the wrong class.
Error in DAGNetwork/calculatePredict (line 13)
Y = predictSingle( ...
Error in DAGNetwork/predict (line 125)
Y = this.calculatePredict( ...
The detector after training can be used without problem.
Did I miss something? Are the checkpoint models of same class as final detector itself?
Thanks
  2 commentaires
Guillaume PERRIN
Guillaume PERRIN le 29 Mar 2020
Modifié(e) : Guillaume PERRIN le 29 Mar 2020
Hi, indeed checkpoints that are associated to some DAGNetwork architectures cannot be used to make predictions. As an example, those who contain batchNorm layers. See https://fr.mathworks.com/matlabcentral/answers/423588-how-to-classify-with-dag-network-from-checkpoint or https://fr.mathworks.com/matlabcentral/answers/451383-issue-with-batch-normalization-layer-of-saved-cnn#answer_366855.
For a workaround, you can still retrain from your checkpoint during 1 epoch, with a minimalist training set, at a very low learning rate.
Best,
Guillaume
Song Decn
Song Decn le 8 Fév 2021
Thanks ! Guillaume

Connectez-vous pour commenter.

Réponses (0)

Catégories

En savoir plus sur Deep Learning Toolbox dans Help Center et File Exchange

Produits


Version

R2019b

Community Treasure Hunt

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

Start Hunting!

Translated by