Customising your models with TensorFlow 2