➜  tensorflow-data git:(master) ✗ python train.py
_________________________________________________________________
Layer (type)                 Output Shape              Param #
=================================================================
input_1 (InputLayer)         (None, 334, 334, 3)       0
_________________________________________________________________
conv2d_1 (Conv2D)            (None, 332, 332, 32)      896
_________________________________________________________________
max_pooling2d_1 (MaxPooling2 (None, 166, 166, 32)      0
_________________________________________________________________
conv2d_2 (Conv2D)            (None, 164, 164, 64)      18496
_________________________________________________________________
max_pooling2d_2 (MaxPooling2 (None, 82, 82, 64)        0
_________________________________________________________________
conv2d_3 (Conv2D)            (None, 80, 80, 128)       73856
_________________________________________________________________
max_pooling2d_3 (MaxPooling2 (None, 40, 40, 128)       0
_________________________________________________________________
conv2d_4 (Conv2D)            (None, 38, 38, 3)         3459
_________________________________________________________________
max_pooling2d_4 (MaxPooling2 (None, 19, 19, 3)         0
_________________________________________________________________
activation_1 (Activation)    (None, 19, 19, 3)         0
_________________________________________________________________
flatten_1 (Flatten)          (None, 1083)              0
=================================================================
Total params: 96,707
Trainable params: 96,707
Non-trainable params: 0
_________________________________________________________________
Epoch 1/1
100/100 [==============================] - 1656s 17s/step - loss: 1685.4183 - acc: 0.9900