Your code in the train_step() function is fairly simple, since it just calls the functions you tested earlier:
You compute a_G using the generated_image.
You compute J_style using compute_style_cost(…)
You compute J_content using compute _content_cost(…)
You compute J using total_cost(…).
Assuming you did not modify any of the provided code, the gradients should be available via the call to tape.gradient().
What function arguments did you use? How can I share it with you? Can I just include my code (4-5 lines) here? I do not want to violate any Coursera rules…
Did you run all of the cells from the top of the notebook… YES
Also, are you running the notebook in Coursera Labs? YES