code: def test_multiple_output_forward(pretrain_file):
      """
      Test a couple different sizes of output layers
      """
      model = build_model(
          pretrain_file,
          "--num_output_layers",
          "1",
          "--num_lstm_layers",
          "2",
      )
      run_forward_checks(model)
  
      model = build_model(
          pretrain_file,
          "--num_output_layers",
          "2",
          "--num_lstm_layers",
          "2",
      )
      run_forward_checks(model)
  
      model = build_model(
          pretrain_file,
          "--num_output_layers",
          "3",
          "--num_lstm_layers",
          "2",
      )
      run_forward_checks(model)
  
imports: import os
  import pytest
  import torch
  from stanza.models.common import pretrain
  from stanza.models.common.utils import set_random_seed
  from stanza.models.constituency import parse_transitions
  from stanza.tests import *
  from stanza.tests.constituency import test_parse_transitions
  from stanza.tests.constituency.test_trainer import build_trainer
label_desc: Use image node_id 16 for calling a global function with example usage: test_multiple_output_forward(pretrain_file) without return types