I didn’t have a look at the notebook. But this is a common error. You should check the input dimensions of the data and the dimension of the input your network expects. A good way to debug this is to print the dimensions of the tensor as training progresses, if the dimension change in the way, you’ll know what to do.