Cannot run example code.

[murtuzamdahod]

I cannot run the example code that is provided in the markdown. I have tried running it with GPU in local and CPU and GPU both in colab. While trying to fit the model it gives AttributeError saying tuple has no attribute shape.