calculate gradients of model (fix code)

Related categories: Python / Tensorflow / Keras / Deep Learning