generalize the first conv layer

marph91 · Jun 9, 2021 · 4a8f74a · 4a8f74a
1 parent e7a12d5
commit 4a8f74a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/playground/05_intro_modified.py b/playground/05_intro_modified.py
@@ -33,7 +33,7 @@
  kernel_quantizer="ste_sign",
  kernel_constraint="weight_clip",
  use_bias=False,
- input_shape=(28, 28, 1),
+ input_shape=train_images.shape[1:],
  )
 )
 # Scale is not needed, since we clip afterwards anyway.