def test_load_save_finetune_model(self): self.assertTrue( test_utils.verify_torchscript_models(self.finetune_model))
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))
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))
def test_load_save_pretrain_model(self): self.assertTrue( test_utils.verify_torchscript_models(self.pretrain_model.model))
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))