conv2d layer dimension compatibility problem

[Jiwoon Lee]

train.py line 55

net.build(input_shape)    # input_shape=[84, 84, 1]

ValueError: Input 0 of layer conv2d is incompatible with the layer: : expected min_ndim=4, found ndim=3. Full shape received: [84, 84, 1]