docstr: Scripted model outputs the same as eager mode
docstr_tok: ["Scripted","model","outputs","the","same","as","eager","mode"]
code: def test_torchscript_infer(self):
      """Scripted model outputs the same as eager mode"""
  
      upsample_scales = [5, 5, 8]
      n_rnn = 128
      n_fc = 128
      n_classes = 128
      hop_length = 200
      n_batch = 2
      n_time = 50
      n_freq = 25
      n_output = 64
      n_res_block = 2
      n_hidden = 32
      kernel_size = 5
  
      model = WaveRNN(
          upsample_scales,
          n_classes,
          hop_length,
          n_res_block,
          n_rnn,
          n_fc,
          kernel_size,
          n_freq,
          n_hidden,
          n_output,
      )
      model.eval()
      x = torch.rand(n_batch, n_freq, n_time)
      torch.random.manual_seed(0)
      out_eager = model.infer(x)
      torch.random.manual_seed(0)
      out_script = torch_script(model).infer(x)
      self.assertEqual(out_eager, out_script)
  
imports: import itertools
  from collections import namedtuple
  import torch
  from parameterized import parameterized
  from torchaudio.models import ConvTasNet, DeepSpeech, Wav2Letter, WaveRNN
  from torchaudio.models.wavernn import MelResNet, UpsampleNetwork
  from torchaudio_unittest import common_utils
  from torchaudio_unittest.common_utils import torch_script
label_desc: Use image node_id 3 for calling the TestWaveRNN obj's underlying member method code with example usage: obj.test_torchscript_infer() without return types