Sathsara_Rasantha:

I kindly request someone to take a look at this issue and solve it for me.


You’ll need to be more specific than that. The more specific you are, the easier it will be for us to understand and suggest solutions.
From the outset, you can try disabling CUDA and first try to make it work end to end on a CPU (with a smaller dataset).