This is the error i receive, i do not know why this happens because the first epoch is running smoothly, i have give the notebook file which contains the whole code. I am doing