message: Moved total loss calculation logic
diff: " # using teacher forcing\n",
  " dec_input = tf.expand_dims(targ[:, t], 1)\n",
  " \n",
  + " batch_loss = (loss / int(targ.shape[1]))\n",
  + " \n",
  + " total_loss += batch_loss\n",
  + " \n",
  " variables = encoder.variables + decoder.variables\n",
  " \n",
  " gradients = tape.gradient(loss, variables)\n",
  " \n",
  " optimizer.apply_gradients(zip(gradients, variables), tf.train.get_or_create_global_step())\n",
  " \n",
  - " batch_loss = (loss / int(targ.shape[1]))\n",
  - " \n",
  - " total_loss += batch_loss\n",
  - " \n",
  " if batch % 100 == 0:\n",
  " print('Epoch {} Batch {} Loss {:.4f}'.format(epoch + 1,\n",
  " batch,\n",
  