class_docstr: The Brain implementation for Tacotron2
class_docstr_tok: ["The","Brain","implementation","for","Tacotron2"]
docstr: At the end of the optimizer step, apply noam annealing.
docstr_tok: ["At","the","end","of","the","optimizer","step",",","apply","noam","annealing","."]
code: def on_fit_batch_end(self, batch, outputs, loss, should_step):
      """At the end of the optimizer step, apply noam annealing."""
      if should_step:
          self.hparams.lr_annealing(self.optimizer)
  
file_path: speechbrain/recipes/LJSpeech/TTS/tacotron2/train.py
imports: import torch
  import speechbrain
  import sys
  import logging
  from hyperpyyaml import load_hyperpyyaml
  from speechbrain.utils.text_to_sequence import text_to_sequence
  from speechbrain.utils.data_utils import scalarize
label_desc: Use image node_id 3 for calling the Tacotron2Brain obj's underlying member method code with example usage: obj.on_fit_batch_end(batch, outputs, loss, should_step) without return types