I am working on a project where I need to create a custom TensorFlow model but I don\'t need the built-in fit function, or basically anything else except the
fit