docstr: Computes the forward pass
  Arguments
  ---------
  batch: str
      a single batch
  stage: speechbrain.Stage
      the training stage
  Returns
  -------
  the model output
docstr_tok: ["Computes","the","forward","pass","Arguments","--","--","--","--","-","batch",":","str","a","single","batch","stage",":","speechbrain.Stage","the","training","stage","Returns","--","--","--","-","the","model","output"]
code: def compute_forward(self, batch, stage):
      """Computes the forward pass
      Arguments
      ---------
      batch: str
          a single batch
      stage: speechbrain.Stage
          the training stage
      Returns
      -------
      the model output
      """
      inputs, _ = self.batch_to_device(batch)
      return self.hparams.model(*inputs)
  
imports: import os
  import sys
  import torch
  import logging
  import torchaudio
  import numpy
  import speechbrain
  from speechbrain.inference.vocoders import HIFIGAN
  from pathlib import Path
  from hyperpyyaml import load_hyperpyyaml
  from speechbrain.utils.data_utils import scalarize
label_desc: Use image node_id 2 for calling the FastSpeech2Brain obj's underlying member method code with example usage: obj.compute_forward(batch, stage) and returns: self