Hello When I try to run the training loop I get the following error. Could you help me
/usr/local/lib/python3.7/dist-packages/tensorflow/python/framework/func_graph.py in autograph_handler(*args, **kwargs)
1123 recursive=True,
1124 optional_features=autograph_options,
→ 1125 user_requested=True,
1126 ))
1127 except Exception as e: # pylint:disable=broad-except
TypeError: in user code:
File "<ipython-input-63-38c196ad1363>", line 44, in train_step_fn *
prediction_dict = model.predict(preprocessed_image_tensor, true_shape_tensor)
TypeError: apply_gradients() missing 1 required positional argument: 'grads_and_vars'