Python verify_torchscript_modelsの例

コード例 #1

ファイル: test_mmbt.py プロジェクト: yyht/Transformer-MM-Explainability

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

コード例 #2

ファイル: test_mmf_transformer.py プロジェクト: 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))

コード例 #3

ファイル: test_vilbert.py プロジェクト: 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))

コード例 #4

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

コード例 #5

ファイル: test_vilbert.py プロジェクト: 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))