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