question: Getting error after completion of 1st epoch
context: I am training an image classifier on tpu and am getting errors after execution of the first epoch.
  https://colab.research.google.com/drive/1Lgz0mF6UiLirsDltPQH5HtctmVvb7gHm?usp=sharing
answer: I don't see an error in the provided link.
  Is this still relevant?