Train model function is giving errors

i have attached my google colab notebook for your reference.

https://colab.research.google.com/drive/1gPvaw6oby_ha2n0OxKZNbk-CVlrzISHa?usp=sharing

sir i am facing errors in this

hey @Aashi-Agarwal-2380268668952871 ,
i have replied regarding the same in your another doubt.
Kindly have a look at that.

Thank You :slightly_smiling_face:.