Training network after combining two imds
조회 수: 5 (최근 30일)
이전 댓글 표시
Hi all,
I'm trying to use a combined dataset to train my network
imds and augImds worked great.
Then I
combine(imds, augImds);
The problem raises when I do
net = trainNetwork(combinedImds, layers, options);
The error pops-up:
Error using trainNetwork: Invalid training data. The output size [1 1 1 2] of the last layer does not match the response size [0 0 1 1].
The thing is that the training works when using imds, or augImds as arguments. But not for combineImds.
Does anyone know how to solve it?
By the way, I haven't found any documentation explaining what are the meaning of these elements in output and response size. My classes are 2, so I wonder the last element for output size represents the # classes. But what are the other elements?
Thanks
댓글 수: 0
답변 (0개)
참고 항목
카테고리
Help Center 및 File Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!