x_train = x_train.astype('float32') / 255.
x_test = x_test.astype('float32') / 255.