Error when running in Colab

hello everyone,
I have done the test, submitted (Course 4 W2A2) and then copy pasted the code on colab (using the files available). Works fine although I do have an error on colab that I can’t fix:

that leads to a network having 154 layers. At the end, the accuracy is not exactly the same, but for the purpose of the exercise, it works fine. I am just curious to understand why this error. Could it be another version of tf?

Hello @pat!

Yes, this is because of TF version. Coursera use 2.1 I think and Colab use 2.9.