layer_dense(units = 256, activation = 'relu', #Fully connected units input_shape = c(28*28)) %>% layer_dense(units = 128, activation = 'relu') %>% layer_dense(units = 10, activation