ageron / ageron/handson-ml2

Chapter 19 - Conv2d dimensions.

Abierto
#176 0 comentarios 0 reacciones 0 asignados Ver en GitHub
Lenguaje dominante
Jupyter Notebook
Estrellas
30k
Forks
13.1k
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

With

dataset = tf.data.Dataset.from_tensor_slices((X_train[..., np.newaxis],y_train)).repeat().batch(batch_size)

I get:

ValueError: Error when checking input: expected conv2d_input to have 4 dimensions, but got array with shape (None, 28, 28, 1, 1)

Replacing with:

dataset = tf.data.Dataset.from_tensor_slices((X_train, y_train)).repeat().batch(batch_size)

allows the training to run.

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.