class ConformerTestClass(unittest.TestCase): def setUp(self): self.conformer = Conformer() def test_conformer_add_method_return_correct_result(self): result = self.conformer.add(2, 2) self.assertEqual(4, result) def main(): unittest.main()
def Conformer_base_patch16(pretrained=False, **kwargs): model = Conformer(patch_size=16, channel_ratio=6, embed_dim=576, depth=12, num_heads=9, mlp_ratio=4, qkv_bias=True, **kwargs) if pretrained: raise NotImplementedError return model
def Conformer_small_patch32(pretrained=False, **kwargs): model = Conformer(patch_size=32, channel_ratio=4, embed_dim=384, depth=12, num_heads=6, mlp_ratio=4, qkv_bias=True, **kwargs) if pretrained: raise NotImplementedError return model
def setUp(self): self.conformer = Conformer()