search

Waikato / wekaDeeplearning4j

Weka package for the Deeplearning4j java library
https://deeplearning.cms.waikato.ac.nz/
GNU General Public License v3.0
185 stars 202 forks source link

Problem For Label classes inferred using label generator #62

Closed trainingemad closed 3 years ago

trainingemad commented 3 years ago

Description

I have been training a model with 244 label classes; however, an unsupported expected occurs that states Cannot do conversion to one hot using batched reader: 244 output classes, but array.size(1) is 203.

Expected behavior:

BaseImageRecordReader - ImageRecordReader: 244 label classes inferred using label generator ArffMetaDataLabelGenerator

Encountered behavior:

BaseImageRecordReader - ImageRecordReader: 203 label classes inferred using label generator ArffMetaDataLabelGenerator

Additional Information