出现这个问题是因为loss function与label 的encode格式不匹配
第一种情况:
loss=‘sparse_categorical_crossentropy’
label请用noe-hot编码
第二种情况:
loss=‘categorical_crossentropy’
label是整型就可以了。
其中CIFAR-10的label编码默认为整型,对应的loss function 就应该设为categorical_crossentropy;
但如果要用sparse_categorical_crossentropy,那就先对label做一下处理,转为one-hot编码。
Error when checking target: expected softmax to have shape (10,) but got array with shape (1,)
最新推荐文章于 2022-08-12 08:10:50 发布