报错:在这里插入图片描述

在这里插入图片描述

解决方案:

将model.compile里面的loss参数修改如下即可:

 model.compile(optimizer='adam', 
 loss=tf.keras.SparseCaticalCrossentropy(from_logits=True), 
 metrics=['accuracy'])

改为:

 model.compile(optimizer='adam', 
 loss='sparse_categorical_crossentropy',
  metrics=['accuracy'])

这样就能解决了。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐