code: def test_load_save_pretrain_model(self):
      self.model_config["training_head_type"] = "pretraining"
      pretrain_model = build_model(self.model_config)
      self.assertTrue(
          test_utils.verify_torchscript_models(pretrain_model.model)
      )
  
file_path: mmf/tests/models/test_vilbert.py
imports: import gc
  import unittest
  import tests.test_utils
  import torch
  from mmf.modules.hf_layers import undo_replace_with_jit
  from mmf.utils.build import build_model
  from mmf.utils.configuration import Configuration
  from mmf.utils.env import setup_imports, teardown_imports
label_desc: Use image node_id 3 for calling the TestViLBertTorchscript obj's underlying member method code with example usage: obj.test_load_save_pretrain_model() without return types