I am attempting to implement a U-Net with region proposal network in tensorflow. I have a keras model that compiles, but when I try to write a training loop with GradientTap
