Python verify_torchscript_models Examples

Example #1

File: test_mmbt.py Project: yyht/Transformer-MM-Explainability

 def test_load_save_finetune_model(self):
     self.assertTrue(
         test_utils.verify_torchscript_models(self.finetune_model))

Example #2

File: test_mmf_transformer.py Project: SunYanCN/pythia

 def test_load_save_finetune_model(self):
     model = build_model(self.config.model_config[self.model_name])
     self.assertTrue(test_utils.verify_torchscript_models(model))

Example #3

File: test_vilbert.py Project: zhang703652632/mmf

 def test_load_save_finetune_model(self):
     self.model_config["training_head_type"] = "classification"
     finetune_model = build_model(self.model_config)
     self.assertTrue(test_utils.verify_torchscript_models(finetune_model.model))

Example #4

 def test_load_save_pretrain_model(self):
     self.assertTrue(
         test_utils.verify_torchscript_models(self.pretrain_model.model))

Example #5

File: test_vilbert.py Project: zhang703652632/mmf

 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))