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

조회 수: 2 (최근 30일)
Song Decn
Song Decn 2020년 3월 27일
댓글: Song Decn 2021년 2월 8일
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
Guillaume PERRIN
Guillaume PERRIN 2020년 3월 29일
편집: Guillaume PERRIN 2020년 3월 29일
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 2021년 2월 8일
Thanks ! Guillaume

댓글을 달려면 로그인하십시오.

답변 (0개)

카테고리

Help CenterFile Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기

태그

제품


릴리스

R2019b

Community Treasure Hunt

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

Start Hunting!

Translated by