I want to plot the gradients for each epoch and trying to obtain the values using gradient tape. But I am getting the following error.
batch_size = 128 epochs
