Hi @Ejaz_Habib_Dar and @Yan_Du , I had the same error when using a different version of TF (the default version in the colab notebook is TF2.15 which is not compatible with the autograder currently) than the recommended TF 2.8 version. Please check this answer in this post and see if it works for you.